Home
last modified time | relevance | path

Searched defs:run_id (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnccl_utils.cc126 int64_t run_id = -1; member
140 RunId run_id, OpId op_id, NcclCliqueKey clique_key, in AcquireNcclClique()
224 RunId run_id, OpId op_id, std::vector<GlobalDeviceId> participants, in AcquireNcclComm()
Dgpu_executable_run_options.h89 RunId run_id; member
/external/tensorflow/tensorflow/core/tfrt/gpu/
Dgpu_shared_context.cc25 int64_t run_id, in GpuSharedContext()
Dgpu_shared_context.h52 int64_t run_id() const { return run_id_; } in run_id() function
/external/tensorflow/tensorflow/core/protobuf/
Dsnapshot.proto24 string run_id = 2; field
/external/tensorflow/tensorflow/compiler/xla/
Dexecutable_run_options.cc128 RunId ExecutableRunOptions::run_id() const { return run_id_; } in run_id() function in xla::ExecutableRunOptions
/external/tensorflow/tensorflow/core/summary/
Dsummary_db_writer.cc219 GraphDef* graph, uint64 now, int64_t run_id, in Save()
335 Status SaveGraph(int64_t run_id) { in SaveGraph()
400 int64_t run_id() TF_LOCKS_EXCLUDED(mu_) { in run_id() function in tensorflow::__anon5feea5b20111::RunMetadata
408 int64_t run_id; in SetGraph() local
899 int64_t run_id = meta_.run_id(); in ~SummaryDbWriter() local
Dsummary_db_writer_test.cc196 int64_t run_id = QueryInt("SELECT run_id FROM Runs"); in TEST_F() local
/external/python/mobly/mobly/
Dtest_runner.py274 def run_id(self): member in TestRunner._TestRunMetaData
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_runner.cc235 RunId run_id; in ExecuteReplicatedImpl() local
474 RunId run_id) { in GetServiceRunOptionsForDevice()
Dcollective_ops_utils.h207 RunId run_id; member
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dio_ops.cc85 auto run_id = random::New64(); in DoCompute() local
198 uint64 run_id, in WriteMetadataFile()
470 Status WriteMetadataFile(Env* env, const std::string& path, uint64 run_id, in WriteMetadataFile()
Dsnapshot_dataset_op.cc1190 std::string run_id = ""; in InitializeIterator() local
1249 const string& run_id, int64_t version) in SnapshotReaderIterator()
1424 tstring hash_dir, run_id, run_dir; in RestoreInternal() local
1663 const string& run_id) in SnapshotWriterIterator()
/external/tensorflow/tensorflow/compiler/xrt/
Dxrt.proto122 int64 run_id = 3; field
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.cc558 int partition, const RunId& run_id) { in ExecuteHelper()
726 RunId run_id; in ExecuteOnLocalDevices() local
/external/tensorflow/tensorflow/core/data/
Dsnapshot_utils.cc73 std::string RunDirectory(const std::string& hash_directory, uint64 run_id) { in RunDirectory()
80 const std::string& run_id) { in RunDirectory()
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_stream_executor_client.cc1901 int executable_idx, const RunId& run_id, const ExecuteOptions& options, in EnqueueExecution()
2093 const RunId& run_id, const ExecuteOptions& options, bool fill_future, in ExecuteHelper()
2203 RunId run_id; in Execute() local
Dtfrt_cpu_pjrt_client.cc1305 const RunId& run_id, const ExecuteOptions& options, in ExecuteHelper()
1623 RunId run_id; in Execute() local
/external/tensorflow/tensorflow/compiler/jit/kernels/
Dxla_ops.cc369 xla::RunId run_id(0); in ComputeAsync() local
/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
Dc_api_decl.h166 int64_t run_id; member