- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for EncodePrimitiveTypeAsDataType (0.19 sec)
-
tensorflow/compiler/jit/xla_host_recv_device_context.cc
namespace tensorflow { void XlaHostRecvDeviceContext::CopyDeviceTensorToCPU( const Tensor* device_tensor, StringPiece tensor_name, Device* device, Tensor* cpu_tensor, StatusCallback done) { DataType dtype = EncodePrimitiveTypeAsDataType(shape_.element_type()).value(); TensorShape tensor_shape; Status status = XLAShapeToTensorShape(shape_, &tensor_shape); if (!status.ok()) { done(status); return; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 1.9K bytes - Viewed (0)