Home
last modified time | relevance | path

Searched refs:SubShapedBuffer (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dshaped_buffer.h119 StatusOr<ShapedBuffer> SubShapedBuffer(const ShapeIndex& index) const;
Dshaped_buffer.cc70 StatusOr<ShapedBuffer> ShapedBuffer::SubShapedBuffer( in SubShapedBuffer() function in xla::ShapedBuffer
Dshaped_buffer_test.cc167 auto ssb_statusor = sb.SubShapedBuffer({1}); in TEST()