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.cc57 for (auto& index_to_buffer : shaped_buffer.buffers()) { in AllocateShapedBuffer() local
59 xla::ShapeUtil::GetSubshape(on_device_shape, index_to_buffer.first); in AllocateShapedBuffer()
66 index_to_buffer.second = buffer.Forget(); in AllocateShapedBuffer()
/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_state.cc111 for (auto& index_to_buffer : (*buffer)->buffers()) { in AllocateScopedShapedBuffer()
113 xla::ShapeUtil::GetSubshape(on_device_shape, index_to_buffer.first); in AllocateScopedShapedBuffer()
119 index_to_buffer.second = buffer.Forget(); in AllocateScopedShapedBuffer()
120 VLOG(2) << "Allocated buffer at " << index_to_buffer.second.opaque() in AllocateScopedShapedBuffer()
121 << " index " << index_to_buffer.first.ToString(); in AllocateScopedShapedBuffer()