Searched defs:logical_on_device_shape (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ | ||
| D | tfrt_cpu_pjrt_client.cc | 903 StatusOr<Shape> TfrtCpuBuffer::logical_on_device_shape() { in logical_on_device_shape() function in xla::TfrtCpuBuffer |
| D | pjrt_stream_executor_client.cc | 581 StatusOr<Shape> PjRtStreamExecutorBuffer::logical_on_device_shape() { in logical_on_device_shape() function in xla::PjRtStreamExecutorBuffer |