Searched defs:on_device_shape (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shaped_buffer.h | 69 const Shape& on_device_shape() const { return on_device_shape_; } in on_device_shape() function
101 void set_shapes(const Shape& on_device_shape) { in set_shapes()
110 void set_shapes(const Shape& on_host_shape, const Shape& on_device_shape) { in set_shapes()
|
| D | shaped_buffer.cc | 34 ShapedBuffer::ShapedBuffer(Shape on_device_shape, int device_ordinal) in ShapedBuffer()
41 ShapedBuffer::ShapedBuffer(Shape on_host_shape, Shape on_device_shape, in ShapedBuffer()
116 ScopedShapedBuffer::ScopedShapedBuffer(Shape on_device_shape, in ScopedShapedBuffer()
123 Shape on_device_shape, in ScopedShapedBuffer()
|
| D | executable.h | 157 ExecutionOutput(Shape on_host_shape, Shape on_device_shape, in ExecutionOutput()
160 ExecutionOutput(Shape on_device_shape, se::DeviceMemoryAllocator* allocator, in ExecutionOutput()
|
| D | transfer_manager.cc | 161 const Shape on_device_shape = HostShapeToDeviceShape(literal.shape()); in TransferArrayToDeviceAsync() local
395 Shape on_device_shape = HostShapeToDeviceShape(on_host_shape); in AllocateScopedShapedBuffer() local
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_state.cc | 87 xla::Shape on_device_shape = transfer_manager->HostShapeToDeviceShape(shape); in AllocateScopedShapedBuffer() local
150 const xla::Shape& on_device_shape) in XRTTupleAllocation()
222 const xla::Shape& on_device_shape, xla::Backend* backend, in CreateFromBuffer()
352 const xla::Shape& XRTTupleAllocation::on_device_shape() const { in on_device_shape() function in tensorflow::XRTTupleAllocation
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_tensor.cc | 46 xla::Shape on_device_shape = in AllocateShapedBuffer() local
|
| D | xla_tpu_device.cc | 87 const xla::Shape& on_device_shape = in TpuPaddedShapeFn() local
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_executable_interface.cc | 144 const Shape& on_device_shape = result.Result().on_device_shape(); in AllocateOutputMemoryWithInputReuse() local
|
| D | c_api_decl.h | 230 XLA_Shape on_device_shape; member
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_stream_executor_client.cc | 390 Shape on_device_shape = dst_buffer.on_device_shape(); in AllocateDestinationBuffer() local
998 Shape on_device_shape, std::shared_ptr<TrackedDeviceBuffer> device_buffer, in PjRtStreamExecutorBuffer()
1492 Shape on_device_shape = ShapeUtil::MakeTupleShape(device_shapes); in MakeTupleHelper() local
|
| D | pjrt_stream_executor_client.h | 469 const Shape& on_device_shape() const override { return on_device_shape_; } in on_device_shape() function
|