Searched defs:argument_handles (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_stream_executor_client.cc | 1799 absl::Span<PjRtBuffer* const> argument_handles, in MakeExecutionInputsAndWaitForEvents()
1854 absl::Span<PjRtBuffer* const> argument_handles, int replica, int partition, in EnqueueExecution()
2046 absl::Span<PjRtBuffer* const> argument_handles, int replica, int partition, in ExecuteHelper()
2146 absl::Span<const std::vector<PjRtBuffer*>> argument_handles, in Execute()
2261 absl::Span<PjRtBuffer* const> argument_handles, PjRtDevice* device, in ExecuteSharded()
2284 absl::Span<PjRtBuffer* const> argument_handles, PjRtDevice* device, in ExecutePortable()
|
| D | tfrt_cpu_pjrt_client.cc | 1340 absl::Span<PjRtBuffer* const> argument_handles, int replica, int partition, in ExecuteHelper()
1615 absl::Span<const std::vector<PjRtBuffer*>> argument_handles, in Execute()
1718 absl::Span<PjRtBuffer* const> argument_handles, PjRtDevice* device, in ExecuteSharded()
1743 absl::Span<PjRtBuffer* const> argument_handles, PjRtDevice* device, in ExecutePortable()
|
| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
| D | tpu_client.cc | 623 absl::Span<PyTpuBuffer* const> argument_handles) { in Execute()
668 absl::Span<const std::vector<PyTpuBuffer*>> argument_handles) { in ExecuteOnLocalDevices()
|