Searched defs:AsShapedBuffer (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ | ||
| D | tracked_device_buffer_test.cc | 51 TEST(TrackedDeviceBufferTest, AsShapedBuffer) { in TEST() argument |
| D | tracked_device_buffer.cc | 120 ShapedBuffer TrackedDeviceBuffer::AsShapedBuffer( in AsShapedBuffer() function in xla::TrackedDeviceBuffer |
| D | pjrt_stream_executor_client.cc | 1267 StatusOr<ShapedBuffer> PjRtStreamExecutorBuffer::AsShapedBuffer() const { in AsShapedBuffer() function in xla::PjRtStreamExecutorBuffer |