Home
last modified time | relevance | path

Searched refs:run_id (Results 1 – 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/
Dcollective_ops_utils.h81 explicit RendezvousKey(const RunId& run_id, in RendezvousKey()
85 : run_id(run_id), in RendezvousKey()
93 return H::combine(std::move(h), k.run_id, k.global_devices, in AbslHashValue()
98 return a.run_id == b.run_id && a.global_devices == b.global_devices &&
111 run_id.ToString(), GlobalDeviceIdsToString(global_devices), in ToString()
115 RunId run_id; member
Dhlo_runner.cc223 RunId run_id; in ExecuteReplicatedImpl() local
231 device, streams.back().get(), device_assignment, run_id)); in ExecuteReplicatedImpl()
444 RunId run_id) { in GetServiceRunOptionsForDevice() argument
454 run_options.set_run_id(run_id); in GetServiceRunOptionsForDevice()
Dhlo_runner.h150 RunId run_id);
/external/autotest/server/cros/
Dgoofy_client.py299 def _get_test_run_info(self, run_id): argument
306 return self._client.GetTestRunStatus(run_id)
309 def _wait_on_run(self, run_id): argument
318 run_info = self._get_test_run_info(run_id)
328 run_info = self._get_test_run_info(run_id)
348 run_id = self._client.RunTest(suite_name)
349 self._wait_on_run(run_id)
350 return self._get_test_run_info(run_id)
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcollective_permute_thunk.cc69 RunId run_id; member
74 run_id.ToString(), num_participants); in ToString()
79 return H::combine(std::move(h), k.run_id); in AbslHashValue()
82 return a.run_id == b.run_id; in operator ==()
236 RendezvousKey key{params.run_id, params.device_assn->replica_count()}; in ExecuteOnStream()
Dnccl_collective_thunk.cc89 RendezvousKey rendezvous_key(params.run_id, std::move(participants), in ExecuteOnStream()
Dthunk.h108 RunId run_id; member
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dio_ops.cc52 auto run_id = random::New64(); in DoCompute() local
53 auto run_dir = snapshot_util::RunDirectory(path, run_id); in DoCompute()
56 WriteMetadataFile(ctx->env(), path, run_id, dataset->output_dtypes(), in DoCompute()
66 TF_RETURN_IF_ERROR(WriteMetadataFile(ctx->env(), path, run_id, in DoCompute()
162 uint64 run_id, in WriteMetadataFile() argument
168 strings::Printf("%llu", static_cast<unsigned long long>(run_id))); in WriteMetadataFile()
293 dataset()->metadata_.run_id()); in InitializeInput()
Dsnapshot_dataset_op.cc568 auto run_dir = snapshot_util::RunDirectory(hash_dir, metadata.run_id()); in Initialize()
1285 std::string run_id = ""; in InitializeIterator() local
1289 run_id = "custom"; in InitializeIterator()
1297 hash_dir_, run_id); in InitializeIterator()
1300 if (run_id.empty() && metadata.run_id().empty()) { in InitializeIterator()
1304 if (run_id.empty()) { in InitializeIterator()
1305 run_id = metadata.run_id(); in InitializeIterator()
1325 hash_dir_, run_id, metadata.version()); in InitializeIterator()
1344 const string& run_id, int64 version) in SnapshotReaderIterator() argument
1347 run_id_(run_id), in SnapshotReaderIterator()
[all …]
Dio_ops.h56 Status WriteMetadataFile(Env* env, const std::string& path, uint64 run_id,
Dsnapshot_util.h60 std::string RunDirectory(const std::string& hash_directory, uint64 run_id);
62 const std::string& run_id);
Dsnapshot_util.cc57 std::string RunDirectory(const std::string& hash_directory, uint64 run_id) { in RunDirectory() argument
60 strings::Printf("%llu", static_cast<unsigned long long>(run_id))); in RunDirectory()
64 const std::string& run_id) { in RunDirectory() argument
65 return io::JoinPath(hash_directory, run_id); in RunDirectory()
/external/tensorflow/tensorflow/core/summary/
Dsummary_db_writer.cc219 GraphDef* graph, uint64 now, int64 run_id, int64* graph_id) in Save() argument
226 TF_RETURN_WITH_CONTEXT_IF_ERROR(saver.SaveGraph(run_id), "SaveGraph"); in Save()
334 Status SaveGraph(int64 run_id) { in SaveGraph() argument
345 if (run_id != kAbsent) insert.BindInt(1, run_id); in SaveGraph()
399 int64 run_id() TF_LOCKS_EXCLUDED(mu_) { in run_id() function in tensorflow::__anon4ebb98140111::RunMetadata
407 int64 run_id; in SetGraph() local
411 run_id = run_id_; in SetGraph()
416 GraphWriter::Save(db, &txn, ids_, g.get(), now, run_id, &graph_id)); in SetGraph()
897 int64 run_id = meta_.run_id(); in ~SummaryDbWriter() local
898 if (run_id == kAbsent) return; in ~SummaryDbWriter()
[all …]
Dsummary_db_writer_test.cc196 int64 run_id = QueryInt("SELECT run_id FROM Runs"); in TEST_F() local
200 EXPECT_LT(0LL, run_id); in TEST_F()
215 EXPECT_EQ(run_id, QueryInt("SELECT run_id FROM Tags")); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/
Dexecutable_run_options.cc128 RunId ExecutableRunOptions::run_id() const { return run_id_; } in run_id() function in xla::ExecutableRunOptions
Dexecutable_run_options.h141 RunId run_id() const;
/external/tensorflow/tensorflow/python/debug/lib/
Ddebug_events_reader.py130 num_no_id = len([run_id for run_id in run_ids if not run_id])
134 for metadata_path, run_id in zip(metadata_paths, run_ids)
135 if not run_id
/external/tensorflow/tensorflow/core/protobuf/
Dsnapshot.proto22 string run_id = 2; field
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_stream_executor_client.cc1683 int executable_idx, const RunId& run_id, const ExecuteOptions& options, in EnqueueExecution() argument
1693 run_id.ToInt()); in EnqueueExecution()
1755 run_options.set_run_id(run_id); in EnqueueExecution()
1859 const RunId& run_id, const ExecuteOptions& options, in ExecuteHelper() argument
1890 argument_handles, replica, partition, executable_idx, run_id, options, in ExecuteHelper()
1949 RunId run_id; in Execute() local
1952 tensorflow::profiler::ContextType::kPjRt, run_id.ToInt()); in Execute()
1976 ExecuteHelper(argument_handles[0], replica, partition, run_id, options); in Execute()
1992 run_id, options); in Execute()
Dpjrt_stream_executor_client.h690 int partition, int executable_idx, const RunId& run_id,
703 int partition, const RunId& run_id, const ExecuteOptions& options,
/external/tensorflow/tensorflow/core/tpu/
Dtpu_on_demand_compiler.cc68 se_options.run_id = options.run_options().run_id().ToInt(); in ToC()
/external/tensorflow/tensorflow/stream_executor/tpu/
Dc_api_decl.h169 int64_t run_id; member
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_execute_op.cc316 if (config.run_id() != 0) { in RunExecutable()
317 run_options.set_run_id(xla::RunId(config.run_id())); in RunExecutable()
/external/autotest/server/cros/faft/
Dfirmware_test.py148 self.run_id = str(uuid.uuid4())
161 logging.info('FirmwareTest initialize begin (id=%s)', self.run_id)
275 logging.info('FirmwareTest initialize done (id=%s)', self.run_id)
357 logging.info('FirmwareTest cleaning up (id=%s)', self.run_id)
392 logging.info('FirmwareTest cleanup done (id=%s)', self.run_id)
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.cc547 int partition, const RunId& run_id) { in ExecuteHelper() argument
716 RunId run_id; in ExecuteOnLocalDevices() local
718 replica, partition, run_id); in ExecuteOnLocalDevices()

12