Home
last modified time | relevance | path

Searched refs:set_buffers (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dshaped_buffer.h92 void set_buffers(ShapeTree<se::DeviceMemoryBase> buffers) { in set_buffers() function
Dshaped_buffer.cc77 sub_shaped_buffer.set_buffers(std::move(sub_buffers)); in SubShapedBuffer()
/external/tensorflow/tensorflow/stream_executor/tpu/
Dc_api_conversions.cc39 xla_shaped_buffer.set_buffers(xla_shape_tree); in FromC()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_execute_op.cc176 shaped_buffer.set_buffers(buffers.Map<se::DeviceMemoryBase>( in ToShapedBuffer()