Home
last modified time | relevance | path

Searched refs:index_to_buffer (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dxla_tensor.cc53 for (auto& index_to_buffer : shaped_buffer.buffers()) { in AllocateShapedBuffer() local
55 xla::ShapeUtil::GetSubshape(on_device_shape, index_to_buffer.first); in AllocateShapedBuffer()
63 index_to_buffer.second = buffer.Release(); in AllocateShapedBuffer()
/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_state.cc97 for (auto& index_to_buffer : (*buffer)->buffers()) { in AllocateScopedShapedBuffer()
99 xla::ShapeUtil::GetSubshape(on_device_shape, index_to_buffer.first); in AllocateScopedShapedBuffer()
105 index_to_buffer.second = buffer.Release(); in AllocateScopedShapedBuffer()
106 VLOG(2) << "Allocated buffer at " << index_to_buffer.second.opaque() in AllocateScopedShapedBuffer()
107 << " index " << index_to_buffer.first.ToString() << " (" << size in AllocateScopedShapedBuffer()