/external/tensorflow/tensorflow/compiler/jit/ |
D | xla_tensor.h | 63 const xla::ShapedBuffer& shaped_buffer() const { in shaped_buffer() function 67 xla::ShapedBuffer& shaped_buffer() { in shaped_buffer() function 72 void set_shaped_buffer(xla::ScopedShapedBuffer shaped_buffer) { in set_shaped_buffer()
|
D | xla_tensor.cc | 54 xla::ScopedShapedBuffer shaped_buffer(on_host_shape, on_device_shape, in AllocateShapedBuffer() local
|
D | xla_device.cc | 129 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() 67 for (const auto& shaped_buffer : replicated_buffers) { in RegisterInternal() local 78 for (auto& shaped_buffer : replicated_buffers) { in RegisterInternal() local 111 for (const auto& shaped_buffer : replicated_buffers) { in Unregister() local 131 for (auto& shaped_buffer : it->second) { in Unregister() local 145 const ShapedBuffer* shaped_buffer = replicated_buffers[0]; in DeconstructTuple() local 208 for (const auto& shaped_buffer : it->second) { in ResolveInternal() local
|
D | shaped_buffer.cc | 127 ScopedShapedBuffer::ScopedShapedBuffer(ShapedBuffer shaped_buffer, in ScopedShapedBuffer() 150 ShapedBuffer shaped_buffer(static_cast<ShapedBuffer&&>(*this)); in release() local
|
D | transfer_manager.cc | 169 ShapedBuffer shaped_buffer(/*on_host_shape=*/literal.shape(), on_device_shape, in TransferArrayToDeviceAsync() local 194 ShapedBuffer shaped_buffer(/*on_host_shape=*/shape, shape, in TransferArrayFromDevice() local 325 ScopedShapedBuffer shaped_buffer(on_host_shape, std::move(on_device_shape), in AllocateScopedShapedBuffer() local
|
D | executable.cc | 48 const ShapedBuffer& shaped_buffer) { in MakeMaybeOwningDeviceMemoryTree()
|
D | shaped_buffer_test.cc | 185 xla::ScopedShapedBuffer shaped_buffer(shape, shape, /*allocator=*/&allocator, in BM_TakeSubTree() local
|
D | service.cc | 247 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/python/ |
D | shared_device_buffer.cc | 96 ScopedShapedBuffer* shaped_buffer, in FromScopedShapedBuffer() 169 ShapedBuffer shaped_buffer(on_host_shape, on_device_shape, platform, in AsShapedBuffer() local
|
D | shared_device_buffer_test.cc | 106 ShapedBuffer shaped_buffer = abc_tuple_buffer->AsShapedBuffer( in TEST() local
|
/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | buffer_donation_test.cc | 87 auto shaped_buffer = scoped_shaped_buffer.release(); in RunAndCheck() local
|
D | local_client_test_base.cc | 146 const ShapedBuffer& shaped_buffer) { in ShapedBufferToLiteral()
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | gpu_executable.cc | 375 ScopedShapedBuffer shaped_buffer(root->shape(), root->shape(), in ExecuteAsyncOnStream() local
|
/external/tensorflow/tensorflow/compiler/xla/client/ |
D | local_client.cc | 383 const ShapedBuffer& shaped_buffer) { in ShapedBufferToLiteral()
|
/external/tensorflow/tensorflow/compiler/xrt/kernels/ |
D | xrt_execute_op.cc | 167 auto shaped_buffer = run_result.release(); in RunExecutable() local
|