| /external/tensorflow/tensorflow/core/profiler/convert/ |
| D | xplane_to_op_stats_test.cc | 98 constexpr int64 kStepId = 0; in TEST() local
108 10, 90, {{StatType::kStepId, kStepId}}); in TEST()
113 {{StatType::kStepId, kStepId}}); in TEST()
128 constexpr int64 kStepId = 0; in TEST() local
139 10, 90, {{StatType::kStepId, kStepId}}); in TEST()
144 {{StatType::kStepId, kStepId}}); in TEST()
198 constexpr int64 kStepId = 456; in BuildAndStoreXSpaceForTest() local
208 10, 90, {{StatType::kStepId, kStepId}}); in BuildAndStoreXSpaceForTest()
213 {{StatType::kStepId, kStepId}}); in BuildAndStoreXSpaceForTest()
|
| D | xplane_to_step_events_test.cc | 56 10, 90, {{StatType::kStepId, kFirstStepId}}); in TEST()
60 310, 90, {{StatType::kStepId, kSecondStepId}}); in TEST()
65 {{StatType::kStepId, kFirstStepId}}); in TEST()
70 {{StatType::kStepId, kSecondStepId}}); in TEST()
|
| D | xplane_to_tf_data_stats_test.cc | 48 100000000, {{StatType::kStepId, kPrefetchIteratorId}}); in TEST()
53 200000000, 20000000, {{StatType::kStepId, kPrefetchIteratorId}}); in TEST()
65 {{StatType::kStepId, kRangeIteratorId}, in TEST()
73 {{StatType::kStepId, kRangeIteratorId}, in TEST()
|
| D | xplane_to_memory_profile_test.cc | 50 {StatType::kStepId, int64{-93746}}, in TEST()
66 {StatType::kStepId, int64{0}}, in TEST()
81 {StatType::kStepId, int64{-93746}}, in TEST()
|
| D | xplane_to_tf_data_stats.cc | 98 auto iterator_id_stat = iterator_event_visitor.GetStat(StatType::kStepId); in ProcessEventForest()
120 auto iterator_id_stat = iterator_event_visitor.GetStat(StatType::kStepId); in ProcessEventForest()
158 auto iterator_id_stat = visitor.GetStat(StatType::kStepId); in ProcessIteratorEvent()
|
| /external/tensorflow/tensorflow/core/profiler/utils/ |
| D | group_events_test.cc | 42 constexpr int64 kStepId = 0; in TEST() local
54 10, 90, {{StatType::kStepId, kStepId}}); in TEST()
59 {{StatType::kStepId, kStepId}}); in TEST()
86 constexpr int64 kStepId = 0; in TEST() local
97 10, 90, {{StatType::kStepId, kStepId}}); in TEST()
102 {{StatType::kStepId, kStepId}}); in TEST()
128 constexpr int64 kStepId = 0; in TEST() local
139 {{StatType::kStepId, kStepId}, {StatType::kIterNum, kIterNum}}); in TEST()
142 {{StatType::kStepId, kStepId}, {StatType::kIterNum, kIterNum}}); in TEST()
187 {{StatType::kStepId, kSecondStepId}, in TEST()
[all …]
|
| D | group_events.cc | 181 absl::optional<XStatVisitor> stat = event.GetStat(StatType::kStepId); in GetLegacyProducerContext()
226 absl::optional<XStatVisitor> stat = event.GetStat(StatType::kStepId); in GetLegacyConsumerContext()
684 executor_event->GetContextStat(StatType::kStepId); in ProcessTensorFlowLoop()
858 consume_iterator->GetEventVisitor().GetStat(StatType::kStepId); in ConnectTfDataEvents()
888 {StatType::kStepId, StatType::kIterNum}}, in CreateInterThreadConnectInfoList()
891 {StatType::kStepId, StatType::kIterNum}}, in CreateInterThreadConnectInfoList()
|
| D | xplane_schema.h | 125 kStepId, enumerator
|
| D | xplane_schema.cc | 135 {"id", kStepId}, in GetStatTypeMap()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_test.cc | 50 static constexpr int kStepId = 10; variable
127 &col_exec_mgr_, /*remote_access=*/nullptr, kStepId, dev_mgr_.get(), in Init()
285 params->step_id = kStepId; in PrepareDeviceContext()
332 /*OpKernelContext=*/&ctx, &op_params, col_params_, exec_key, kStepId, in RunReduce()
364 /*OpKernelContext=*/&ctx, &op_params, col_params_, exec_key, kStepId, in RunBroadcast()
405 /*OpKernelContext=*/&ctx, &op_params, col_params_, exec_key, kStepId, in RunGather()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | permuter_test.cc | 43 static int64 kStepId = 123; variable
170 rma_ = new FailTestRMA(dev_mgr_.get(), dev_resolver_.get(), kStepId, in Init()
172 col_exec_ = new BaseCollectiveExecutor(&col_exec_mgr_, rma_, kStepId, in Init()
404 kStepId, &tensor_input_, &tensor_output_); in DoPermute()
428 int64 step_id_ = kStepId;
|
| D | ring_gatherer_test.cc | 114 static int64 kStepId = 123; variable
180 rma_ = new FailTestRMA(dev_mgr_.get(), dev_resolver_.get(), kStepId, in Init()
182 col_exec_ = new BaseCollectiveExecutor(&col_exec_mgr_, rma_, kStepId, in Init()
452 op_params.step_id = kStepId; in DoGather()
493 kStepId, &input_tensor_, output_tensor_ptr); in DoGather()
|
| D | ring_reducer_test.cc | 137 static int64 kStepId = 123; variable
203 rma_ = new FailTestRMA(dev_mgr_.get(), dev_resolver_.get(), kStepId, in Init()
205 col_exec_ = new BaseCollectiveExecutor(&col_exec_mgr_, rma_, kStepId, in Init()
484 op_params.step_id = kStepId; in DoReduce()
526 kStepId, &tensor_, &tensor_); in DoReduce()
|
| D | collective_rma_local_test.cc | 35 static const int kStepId = 123; variable
54 drl_.get(), kStepId); in CollectiveRemoteAccessLocalTest()
|
| D | hierarchical_tree_broadcaster_test.cc | 43 static int64 kStepId = 123; variable
263 rma_ = new FailTestRMA(dev_mgr_.get(), dev_resolver_.get(), kStepId, in Init()
265 col_exec_ = new BaseCollectiveExecutor(&col_exec_mgr_, rma_, kStepId, in Init()
687 kStepId, input_tensor_ptr, output_tensor_ptr); in DoBroadcast()
710 int64 step_id_ = kStepId;
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_rma_distributed_test.cc | 61 static int64 kStepId = 123; variable
70 buf_rendezvous_(kStepId, dev_mgr), in FakeWorker()
215 kStepId, "/job:worker/replica:0/task:0")); in SetUp()
|