Home
last modified time | relevance | path

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

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