Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_state.h291 const xla::Shape on_device_shape_; variable
/external/tensorflow/tensorflow/compiler/xla/service/
Dshaped_buffer.h129 Shape on_device_shape_; variable
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dtfrt_cpu_pjrt_client.h449 const Shape on_device_shape_; variable
Dpjrt_stream_executor_client.h698 const Shape on_device_shape_; variable