Function tvm::runtime::SaveTensor

Function tvm::runtime::SaveTensor#

Function Documentation#

void tvm::runtime::SaveTensor(ffi::TensorView source, support::Stream *strm)#

Save a tensor in the portable CPU tensor format.

Parameters:
  • source – The tensor to save. Non-CPU tensors are copied to the host first.

  • strm – The output stream.