Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dshaped_buffer.h65 const Shape& on_host_shape() const { return on_host_shape_; } in on_host_shape() function
110 void set_shapes(const Shape& on_host_shape, const Shape& on_device_shape) { in set_shapes()
Dshaped_buffer.cc41 ShapedBuffer::ShapedBuffer(Shape on_host_shape, Shape on_device_shape, in ShapedBuffer()
122 ScopedShapedBuffer::ScopedShapedBuffer(Shape on_host_shape, in ScopedShapedBuffer()
Dexecutable.h157 ExecutionOutput(Shape on_host_shape, Shape on_device_shape, in ExecutionOutput()
Dtransfer_manager.cc388 const Shape& on_host_shape, se::DeviceMemoryAllocator* allocator, in AllocateScopedShapedBuffer()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_tensor.cc43 const xla::Shape& on_host_shape, in AllocateShapedBuffer()
/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_state.cc149 const xla::Shape& on_host_shape, in XRTTupleAllocation()
221 const xla::ShapedBuffer& shaped_buffer, const xla::Shape& on_host_shape, in CreateFromBuffer()
348 const xla::Shape& XRTTupleAllocation::on_host_shape() const { in on_host_shape() function in tensorflow::XRTTupleAllocation
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.h208 const Shape& on_host_shape() const { return on_host_shape_; } in on_host_shape() function
Dtpu_client.cc266 Shape on_host_shape, std::shared_ptr<TpuSharedBuffer> device_buffer, in PyTpuBuffer()
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_stream_executor_client.cc364 const Shape& on_host_shape, PjRtDevice* device, in AllocateDestinationBuffer()