| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_tensor.h | 61 const xla::ShapedBuffer& shaped_buffer() const { in shaped_buffer() function
65 xla::ShapedBuffer& shaped_buffer() { in shaped_buffer() function
70 void set_shaped_buffer(xla::ScopedShapedBuffer shaped_buffer) { in set_shaped_buffer()
|
| D | xla_tensor.cc | 50 xla::ScopedShapedBuffer shaped_buffer(on_host_shape, on_device_shape, in AllocateShapedBuffer() local
|
| D | xla_device.cc | 74 const xla::ShapedBuffer& shaped_buffer = xla_tensor->shaped_buffer(); in DefaultPaddedShapeFn() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | allocation_tracker.cc | 35 ScopedShapedBuffer shaped_buffer, const string& tag) { in Register()
69 for (auto& shaped_buffer : replicated_buffers) { in RegisterInternal() local
102 for (const auto& shaped_buffer : replicated_buffers) { in Unregister() local
122 for (auto& shaped_buffer : it->second) { in Unregister() local
136 const ShapedBuffer* shaped_buffer = replicated_buffers[0]; in DeconstructTuple() local
195 for (const auto& shaped_buffer : it->second) { in ResolveInternal() local
|
| D | shaped_buffer.cc | 129 ScopedShapedBuffer::ScopedShapedBuffer(ShapedBuffer shaped_buffer, in ScopedShapedBuffer()
152 ShapedBuffer shaped_buffer(static_cast<ShapedBuffer&&>(*this)); in release() local
|
| D | transfer_manager.cc | 172 ShapedBuffer shaped_buffer(on_device_shape, in TransferArrayToDeviceAsync() local
196 ShapedBuffer shaped_buffer(shape, stream->parent()->device_ordinal()); in TransferArrayFromDevice() local
398 ScopedShapedBuffer shaped_buffer(std::move(on_device_shape), allocator, in AllocateScopedShapedBuffer() local
|
| D | executable.cc | 65 ShapedBuffer shaped_buffer(input_shape, device_ordinal); in ToShapedBuffer() local
94 const ShapedBuffer& shaped_buffer) { in MakeMaybeOwningDeviceMemoryTree()
|
| D | shaped_buffer_test.cc | 186 xla::ScopedShapedBuffer shaped_buffer(shape, /*allocator=*/&allocator, in BM_TakeSubTree() local
|
| D | service.cc | 248 const ShapedBuffer* shaped_buffer = replicated_buffers[replica]; in ResolveAndValidateArguments() local
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_state.cc | 186 auto shaped_buffer = scoped_buffer->release(); in CreateAndTransfer() local
209 auto shaped_buffer = scoped_buffer->release(); in CreateUninitialized() local
221 const xla::ShapedBuffer& shaped_buffer, const xla::Shape& on_host_shape, in CreateFromBuffer()
235 const xla::ShapedBuffer& shaped_buffer, xla::Backend* backend, in CreateFromBuffer()
317 auto shaped_buffer = scoped_buffer->release(); in SwapIn() local
574 const xla::ShapedBuffer& shaped_buffer, in InitializeFromShapedBuffer()
587 xla::ShapedBuffer shaped_buffer(on_host_shape(), on_device_shape(), in ToShapedBuffer() local
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | tracked_device_buffer.cc | 98 ScopedShapedBuffer* shaped_buffer, in FromScopedShapedBuffer()
122 ShapedBuffer shaped_buffer(on_device_shape, device_ordinal_); in AsShapedBuffer() local
|
| D | pjrt_stream_executor_client.cc | 1236 ShapedBuffer shaped_buffer = device_buffer->AsShapedBuffer(on_device_shape_); in ToLiteral() local
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_execute_op.cc | 174 xla::ShapedBuffer shaped_buffer(std::move(host_shape), buffers.shape(), in ToShapedBuffer() local
467 xla::ScopedShapedBuffer shaped_buffer(device_shape, allocator, in AllocateOutputTensors() local
690 auto shaped_buffer = input_buffers->ToShapedBuffer(std::move(host_shape), in DoWork() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | local_client_test_base.cc | 146 const ShapedBuffer& shaped_buffer) { in ShapedBufferToLiteral()
|
| D | buffer_donation_test.cc | 104 ShapedBuffer shaped_buffer = scoped_shaped_buffer.release(); in RunAndCheck() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | local_client.cc | 416 const ShapedBuffer& shaped_buffer) { in ShapedBufferToLiteral()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_execute_op.cc | 271 xla::ScopedShapedBuffer* shaped_buffer = run_result.MutableResult(); in CreateOutputTuple() local
|