Home
last modified time | relevance | path

Searched defs:host_shape (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dexecutable.h67 ExecutionInput(xla::Shape shape, xla::Shape host_shape) in ExecutionInput()
78 xla::Shape host_shape) in ExecutionInput()
93 const Shape& host_shape() const { in host_shape() function
132 void SetHostShape(xla::Shape host_shape) { in SetHostShape()
Dtransfer_manager.h54 virtual Shape HostShapeToDeviceShape(const Shape& host_shape) const { in HostShapeToDeviceShape()
/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_executable.h57 Shape HostShapeToDeviceShape(const Shape& host_shape) override { in HostShapeToDeviceShape()
Dtpu_op_executable.cc106 const xla::Shape& host_shape) { in HostShapeToDeviceShape()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_reshard_variables_op_util.cc188 const xla::Shape& host_shape = in BuildInputBuffers() local
277 const xla::Shape& host_shape = in UpdateOutputVariables() local
Dtpu_reshard_variables_op.cc167 xla::Shape host_shape(executable->input_shapes(0)); in DoTpuExecute() local
Dtpu_execute_op.cc321 const xla::Shape& host_shape = in BuildComputationInputs() local
659 xla::Shape host_shape(executable.input_shapes(0)); in DoWork() local
/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_state.cc416 se::DeviceMemoryAllocator* allocator, xla::Shape* host_shape, in ExpandTreeOfTuples()
462 xla::Shape host_shape; in MakeTuple() local
/external/tensorflow/tensorflow/core/tpu/
Dtpu_execute.cc106 xla::Shape HostShapeToDeviceShape(const xla::Shape& host_shape) { in HostShapeToDeviceShape()