Searched refs:ToShapedBuffer (Results 1 – 6 of 6) sorted by relevance
253 TF_ASSIGN_OR_RETURN(xla::ShapedBuffer shaped_buffer, ToShapedBuffer()); in StoreToLiteral()272 TF_ASSIGN_OR_RETURN(xla::ShapedBuffer shaped_buffer, ToShapedBuffer()); in WriteLiteral()586 xla::StatusOr<xla::ShapedBuffer> XRTTupleAllocation::ToShapedBuffer() { in ToShapedBuffer() function in tensorflow::XRTTupleAllocation
210 xla::StatusOr<xla::ShapedBuffer> ToShapedBuffer();
99 xla::StatusOr<xla::ShapedBuffer> ToShapedBuffer(
62 StatusOr<ShapedBuffer> ExecutionInput::ToShapedBuffer( in ToShapedBuffer() function in xla::ExecutionInput
170 xla::ShapedBuffer ToShapedBuffer(xla::Shape host_shape, in ToShapedBuffer() function690 auto shaped_buffer = input_buffers->ToShapedBuffer(std::move(host_shape), in DoWork()
253 execution_input->ToShapedBuffer( in UpdateDynamicInputs()