Searched defs:on_host_shape (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | shaped_buffer.h | 64 const Shape& on_host_shape() const { return on_host_shape_; } in on_host_shape() function
109 void set_shapes(const Shape& on_host_shape, const Shape& on_device_shape) { in set_shapes()
|
| H A D | shaped_buffer.cc | 41 ShapedBuffer::ShapedBuffer(Shape on_host_shape, Shape on_device_shape, in ShapedBuffer()
122 ScopedShapedBuffer::ScopedShapedBuffer(Shape on_host_shape, in ScopedShapedBuffer()
|
| H A D | executable.h | 155 ExecutionOutput(Shape on_host_shape, Shape on_device_shape, in ExecutionOutput()
|
| H A D | transfer_manager.cc | 382 const Shape& on_host_shape, se::DeviceMemoryAllocator* allocator, in AllocateScopedShapedBuffer()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/ |
| H A D | xrt_state.cc | 150 const xla::Shape& on_host_shape, in XRTTupleAllocation()
221 const xla::ShapedBuffer& shaped_buffer, const xla::Shape& on_host_shape, in CreateFromBuffer()
349 const xla::Shape& XRTTupleAllocation::on_host_shape() const { in on_host_shape() function in tensorflow::XRTTupleAllocation
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | xla_tensor.cc | 48 xla::Shape on_host_shape = in AllocateShapedBuffer() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
| H A D | tpu_client.h | 235 const Shape& on_host_shape() const { return on_host_shape_; } in on_host_shape() function
|
| H A D | tpu_client.cc | 277 Shape on_host_shape, std::shared_ptr<TpuSharedBuffer> device_buffer, in PyTpuBuffer()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| H A D | pjrt_stream_executor_client.cc | 374 const Shape& on_host_shape, PjRtDevice* device, in AllocateDestinationBuffer()
|