Home
last modified time | relevance | path

Searched defs:AsShapedBuffer (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dtracked_device_buffer_test.cc51 TEST(TrackedDeviceBufferTest, AsShapedBuffer) { in TEST() argument
H A Dtracked_device_buffer.cc122 ShapedBuffer TrackedDeviceBuffer::AsShapedBuffer( in AsShapedBuffer() function in xla::TrackedDeviceBuffer
H A Dtfrt_cpu_pjrt_client.cc845 static ShapedBuffer AsShapedBuffer( in AsShapedBuffer() function
H A Dpjrt_stream_executor_client.cc1396 StatusOr<ShapedBuffer> PjRtStreamExecutorBuffer::AsShapedBuffer() const { in AsShapedBuffer() function in xla::PjRtStreamExecutorBuffer