Searched defs:argument_handles (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| H A D | pjrt_client.h | 1165 absl::Span<const std::vector<PjRtBuffer*>> argument_handles, in Execute()
1187 absl::Span<PjRtBuffer* const> argument_handles, PjRtDevice* device, in ExecuteSharded()
1195 absl::Span<PjRtBuffer* const> argument_handles, PjRtDevice* device, in ExecuteSharded()
1218 absl::Span<PjRtBuffer* const> argument_handles, PjRtDevice* device, in ExecutePortable()
1226 absl::Span<PjRtBuffer* const> argument_handles, PjRtDevice* device, in ExecutePortable()
|
| H A D | pjrt_stream_executor_client.cc | 1845 absl::Span<PjRtBuffer* const> argument_handles, in MakeExecutionInputsAndWaitForEvents()
1900 absl::Span<PjRtBuffer* const> argument_handles, int replica, int partition, in EnqueueExecution()
2092 absl::Span<PjRtBuffer* const> argument_handles, int replica, int partition, in ExecuteHelper()
2196 absl::Span<const std::vector<PjRtBuffer*>> argument_handles, in Execute()
2321 absl::Span<PjRtBuffer* const> argument_handles, PjRtDevice* device, in ExecuteSharded()
2350 absl::Span<PjRtBuffer* const> argument_handles, PjRtDevice* device, in ExecutePortable()
|
| H A D | pjrt_c_api_client.cc | 491 absl::Span<const std::vector<PjRtBuffer*>> argument_handles, in Execute()
543 absl::Span<PjRtBuffer* const> argument_handles, PjRtDevice* device, in ExecuteSharded()
566 absl::Span<PjRtBuffer* const> argument_handles, PjRtDevice* device, in ExecutePortable()
|
| H A D | tfrt_cpu_pjrt_client.cc | 1304 absl::Span<PjRtBuffer* const> argument_handles, int replica, int partition, in ExecuteHelper()
1615 absl::Span<const std::vector<PjRtBuffer*>> argument_handles, in Execute()
1731 absl::Span<PjRtBuffer* const> argument_handles, PjRtDevice* device, in ExecuteSharded()
1762 absl::Span<PjRtBuffer* const> argument_handles, PjRtDevice* device, in ExecutePortable()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
| H A D | tpu_client.cc | 633 absl::Span<PyTpuBuffer* const> argument_handles) { in Execute()
678 absl::Span<const std::vector<PyTpuBuffer*>> argument_handles) { in ExecuteOnLocalDevices()
|
| H A D | tpu_client.h | 390 ExecuteWithToken(absl::Span<PyTpuBuffer* const> argument_handles) { in ExecuteWithToken()
|