Searched defs:argument_handles (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_stream_executor_client.cc | 1637 absl::Span<PjRtBuffer* const> argument_handles, in MakeExecutionInputsAndWaitForEvents()
1682 absl::Span<PjRtBuffer* const> argument_handles, int replica, int partition, in EnqueueExecution()
1858 absl::Span<PjRtBuffer* const> argument_handles, int replica, int partition, in ExecuteHelper()
1943 absl::Span<const std::vector<PjRtBuffer*>> argument_handles, in Execute()
2058 absl::Span<PjRtBuffer* const> argument_handles, PjRtDevice* device, in ExecuteSharded()
2081 absl::Span<PjRtBuffer* const> argument_handles, PjRtDevice* device, in ExecutePortable()
|
| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
| D | tpu_client.cc | 622 absl::Span<PyTpuBuffer* const> argument_handles) { in Execute()
668 absl::Span<const std::vector<PyTpuBuffer*>> argument_handles) { in ExecuteOnLocalDevices()
|