| /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 | 124 const xla::ShapedBuffer& shaped_buffer = xla_tensor->shaped_buffer(); in DefaultPaddedShapeFn() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | allocation_tracker.cc | 34 ScopedShapedBuffer shaped_buffer, const string& tag) { in Register()
66 for (const auto& shaped_buffer : replicated_buffers) { in RegisterInternal() local
77 for (auto& shaped_buffer : replicated_buffers) { in RegisterInternal() local
110 for (const auto& shaped_buffer : replicated_buffers) { in Unregister() local
130 for (auto& shaped_buffer : it->second) { in Unregister() local
144 const ShapedBuffer* shaped_buffer = replicated_buffers[0]; in DeconstructTuple() local
207 for (const auto& shaped_buffer : it->second) { in ResolveInternal() local
|
| D | shaped_buffer.cc | 114 ScopedShapedBuffer::ScopedShapedBuffer(ShapedBuffer shaped_buffer, in ScopedShapedBuffer()
137 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
193 ShapedBuffer shaped_buffer(/*on_host_shape=*/shape, shape, in TransferArrayFromDevice() local
320 ScopedShapedBuffer shaped_buffer(on_host_shape, on_device_shape, allocator, in AllocateScopedShapedBuffer() local
|
| D | shaped_buffer_test.cc | 161 xla::ScopedShapedBuffer shaped_buffer(shape, shape, /*allocator=*/&allocator, in BM_TakeSubTree() local
|
| D | service.cc | 246 const ShapedBuffer* shaped_buffer = replicated_buffers[replica]; in ResolveAndValidateArguments() local
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_state.cc | 200 auto shaped_buffer = scoped_buffer->release(); in CreateAndTransfer() local
210 const xla::ShapedBuffer& shaped_buffer, xla::Backend* backend, in CreateFromBuffer()
229 xla::ShapedBuffer shaped_buffer = ToShapedBuffer(); in ToLiteral() local
500 const xla::ShapedBuffer& shaped_buffer, in InitializeFromShapedBuffer()
510 xla::ShapedBuffer shaped_buffer(on_host_shape(), on_device_shape(), in ToShapedBuffer() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | local_client.cc | 213 const ShapedBuffer& shaped_buffer) { in LiteralFromShapedBuffer()
279 const ShapedBuffer& shaped_buffer) { in ShapedBufferToLiteral()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | local_client_test_base.cc | 140 const ShapedBuffer& shaped_buffer) { in ShapedBufferToLiteral()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_executable.cc | 290 ScopedShapedBuffer shaped_buffer(root->shape(), root->shape(), in ExecuteOnStream() local
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_execute_op.cc | 226 auto shaped_buffer = scoped_buffer.release(); in DoWork() local
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 119 LocalShapedBuffer::LocalShapedBuffer(ScopedShapedBuffer shaped_buffer, in LocalShapedBuffer()
123 const ScopedShapedBuffer* LocalShapedBuffer::shaped_buffer() const { in shaped_buffer() function in xla::swig::LocalShapedBuffer
188 ShapedBuffer shaped_buffer(subshape, subshape, platform, device_ordinal); in DestructureTuple() local
|