Searched refs:tensor_copy (Results 1 – 4 of 4) sorted by relevance
215 TensorProto tensor_copy = input_tensor; in ConvertTensorProto() local216 auto* shape = tensor_copy.mutable_tensor_shape(); in ConvertTensorProto()221 ConvertTensorProto(tensor_copy, builder)); in ConvertTensorProto()
295 at::Tensor tensor_copy = CopyTensor(tensor); in GetDeviceData() local296 device_data = TensorToDataHandle(tensor_copy, device); in GetDeviceData()297 cache->Add(std::move(tensor_copy), device_data); in GetDeviceData()
165 auto tensor_copy = tensor;168 Now `tensor` and `tensor_copy` point to the same data and metadata.