Home
last modified time | relevance | path

Searched defs:argument_handles (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_client.h1165 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()
Dpjrt_c_api_client.cc491 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()
Dpjrt_stream_executor_client.cc1845 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()
Dtfrt_cpu_pjrt_client.cc1304 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()
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.cc633 absl::Span<PyTpuBuffer* const> argument_handles) { in Execute()
678 absl::Span<const std::vector<PyTpuBuffer*>> argument_handles) { in ExecuteOnLocalDevices()
Dtpu_client.h390 ExecuteWithToken(absl::Span<PyTpuBuffer* const> argument_handles) { in ExecuteWithToken()