Home
last modified time | relevance | path

Searched defs:host_shape (Results 1 – 8 of 8) 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.h53 virtual Shape HostShapeToDeviceShape(const Shape& host_shape) const { in HostShapeToDeviceShape()
/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_executable.cc106 Shape TpuExecutable::HostShapeToDeviceShape(const Shape& host_shape) { in HostShapeToDeviceShape()
Dtpu_executable_interface.cc55 const Shape& host_shape, const HloInputOutputAliasConfig& alias_config, in AllocateOutputMemoryWithInputReuse()
/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_state.cc415 se::DeviceMemoryAllocator* allocator, xla::Shape* host_shape, in ExpandTreeOfTuples()
462 xla::Shape host_shape; in MakeTuple() local
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_execute_op.cc321 const xla::Shape& host_shape = in BuildComputationInputs() local
663 xla::Shape host_shape(executable.input_shapes(0)); in DoWork() local
/external/tensorflow/tensorflow/core/tpu/
Dtpu_on_demand_compiler.cc198 Shape HostShapeToDeviceShape(const Shape& host_shape) override { in HostShapeToDeviceShape()
Dtpu_execute.cc106 xla::Shape HostShapeToDeviceShape(const xla::Shape& host_shape) { in HostShapeToDeviceShape()