Searched refs:ToShapedBuffer (Results 1 – 3 of 3) sorted by relevance
229 xla::ShapedBuffer shaped_buffer = ToShapedBuffer(); in ToLiteral()252 ToShapedBuffer()); in WriteLiteral()509 xla::ShapedBuffer XRTTupleAllocation::ToShapedBuffer() { in ToShapedBuffer() function in tensorflow::XRTTupleAllocation
177 xla::ShapedBuffer ToShapedBuffer();
107 input_allocations->emplace_back(tuple->ToShapedBuffer()); in GetComputationInputs()