Searched refs:ExecuteOptions (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_client.h | 396 struct ExecuteOptions { struct
462 const ExecuteOptions& options) = 0;
469 const ExecuteOptions& options) = 0;
476 const ExecuteOptions& options) = 0;
|
| D | pjrt_stream_executor_client.h | 652 const ExecuteOptions& options) override;
656 const ExecuteOptions& options) override;
660 const ExecuteOptions& options) override;
683 int device_ordinal, const ExecuteOptions& options,
691 const ExecuteOptions& options, PjRtDevice* device,
696 int device_ordinal, const ExecuteOptions& options,
703 int partition, const RunId& run_id, const ExecuteOptions& options,
|
| D | gpu_multistream_test.cc | 92 ExecuteOptions options; in TEST()
|
| D | pjrt_stream_executor_client.cc | 1636 int device_ordinal, const ExecuteOptions& options, in MakeExecutionInputsAndWaitForEvents()
1683 int executable_idx, const RunId& run_id, const ExecuteOptions& options, in EnqueueExecution()
1823 int device_ordinal, const ExecuteOptions& options, in MakeOutputBuffers()
1859 const RunId& run_id, const ExecuteOptions& options, in ExecuteHelper()
1944 const ExecuteOptions& options) { in Execute()
2059 const ExecuteOptions& options) { in ExecuteSharded()
2082 const ExecuteOptions& options) { in ExecutePortable()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | py_executable.h | 95 ExecuteOptions options_;
|
| D | outfeed_receiver.cc | 410 ExecuteOptions execute_options; in SendShutdownOutfeedHeader()
|
| D | outfeed_receiver_test.cc | 48 ExecuteOptions execute_options; in CompileAndExecute()
|