Home
last modified time | relevance | path

Searched refs:GetOrCreateStatMetadata (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/core/profiler/internal/gpu/
Dcupti_collector.cc153 *plane->GetOrCreateStatMetadata(GetStatTypeStr(StatType::kDeviceId)), in CreateXEvent()
157 xevent.AddStatValue(*plane->GetOrCreateStatMetadata( in CreateXEvent()
162 xevent.AddStatValue(*plane->GetOrCreateStatMetadata( in CreateXEvent()
164 *plane->GetOrCreateStatMetadata(event.annotation)); in CreateXEvent()
168 *plane->GetOrCreateStatMetadata(GetStatTypeStr(StatType::kNVTXRange)), in CreateXEvent()
169 *plane->GetOrCreateStatMetadata(event.nvtx_range)); in CreateXEvent()
173 *plane->GetOrCreateStatMetadata(GetStatTypeStr(StatType::kContextId)), in CreateXEvent()
198 xevent.AddStatValue(*plane->GetOrCreateStatMetadata(GetStatTypeStr( in CreateXEvent()
201 xevent.AddStatValue(*plane->GetOrCreateStatMetadata( in CreateXEvent()
204 xevent.AddStatValue(*plane->GetOrCreateStatMetadata(GetStatTypeStr( in CreateXEvent()
[all …]
/external/tensorflow/tensorflow/core/profiler/utils/
Ddevice_caps_utils.cc29 xplane.AddStatValue(*xplane.GetOrCreateStatMetadata( in SetDeviceCaps()
32 xplane.AddStatValue(*xplane.GetOrCreateStatMetadata( in SetDeviceCaps()
35 xplane.AddStatValue(*xplane.GetOrCreateStatMetadata( in SetDeviceCaps()
38 xplane.AddStatValue(*xplane.GetOrCreateStatMetadata( in SetDeviceCaps()
42 xplane.AddStatValue(*xplane.GetOrCreateStatMetadata( in SetDeviceCaps()
45 xplane.AddStatValue(*xplane.GetOrCreateStatMetadata( in SetDeviceCaps()
Dxplane_builder_test.cc35 *xplane_builder.GetOrCreateStatMetadata("int stat"), int64{1234}); in TEST()
37 *xplane_builder.GetOrCreateStatMetadata("string stat"), in TEST()
40 *xplane_builder.GetOrCreateStatMetadata("double stat"), 1.0); in TEST()
42 *xplane_builder.GetOrCreateStatMetadata("ref stat"), in TEST()
43 *xplane_builder.GetOrCreateStatMetadata("referenced abc")); in TEST()
Dxplane_builder.h108 AddStatValue(metadata, GetOrCreateStatMetadata(value)); in ParseAndAddStatValue()
154 dst_stat->set_ref_value(GetOrCreateStatMetadata(value).id()); in CopyStatValue()
164 const XStatMetadata& GetOrCreateStatMetadata(absl::string_view value);
322 XStatMetadata* GetOrCreateStatMetadata(int64 metadata_id);
329 XStatMetadata* GetOrCreateStatMetadata(absl::string_view name);
330 XStatMetadata* GetOrCreateStatMetadata(std::string&& name);
331 XStatMetadata* GetOrCreateStatMetadata(const char* name) { in GetOrCreateStatMetadata() function
332 return GetOrCreateStatMetadata(absl::string_view(name)); in GetOrCreateStatMetadata()
347 const XStatMetadata& XStatsBuilder<T>::GetOrCreateStatMetadata( in GetOrCreateStatMetadata() function
349 return *stats_metadata_owner_->GetOrCreateStatMetadata(value); in GetOrCreateStatMetadata()
Dxplane_builder.cc82 XStatMetadata* XPlaneBuilder::GetOrCreateStatMetadata(int64 metadata_id) { in GetOrCreateStatMetadata() function in tensorflow::profiler::XPlaneBuilder
89 return GetOrCreateStatMetadata(++last_stat_metadata_id_); in CreateStatMetadata()
92 XStatMetadata* XPlaneBuilder::GetOrCreateStatMetadata(absl::string_view name) { in GetOrCreateStatMetadata() function in tensorflow::profiler::XPlaneBuilder
101 XStatMetadata* XPlaneBuilder::GetOrCreateStatMetadata(std::string&& name) { in GetOrCreateStatMetadata() function in tensorflow::profiler::XPlaneBuilder
Dderived_timeline.cc72 plane_builder->GetOrCreateStatMetadata(GetStatTypeStr(StatType::kGroupId)) in ProcessTfOpEvent()
161 plane.GetOrCreateStatMetadata(GetStatTypeStr(StatType::kGroupId))->id(); in DeriveEventsFromAnnotations()
163 plane.GetOrCreateStatMetadata(GetStatTypeStr(StatType::kStepName))->id(); in DeriveEventsFromAnnotations()
312 device_event.AddStatValue(*device_plane.GetOrCreateStatMetadata( in DeriveEventsFromHostTrace()
316 *device_plane.GetOrCreateStatMetadata("num_launches"), in DeriveEventsFromHostTrace()
319 *device_plane.GetOrCreateStatMetadata("max_launch_time_us"), in DeriveEventsFromHostTrace()
322 *device_plane.GetOrCreateStatMetadata("avg_launch_time_us"), in DeriveEventsFromHostTrace()
Dtfstreamz_utils.cc99 auto* metadata = xplane.GetOrCreateStatMetadata(stat_name); in SerializeToXPlane()
112 xevent.AddStatValue(*metadata, *xplane.GetOrCreateStatMetadata( in SerializeToXPlane()
Dxplane_test_utils.cc70 plane_builder->GetOrCreateStatMetadata(GetStatTypeStr(stat_type))); in CreateXEvent()
Dxplane_utils.cc241 XStatMetadata* stat_metadata = dst.GetOrCreateStatMetadata(stat.Name()); in MergePlanes()
288 dst_event.AddStat(*dst.GetOrCreateStatMetadata(stat.Name()), in MergePlanes()
Dgroup_events.cc47 builder.GetOrCreateStatMetadata(GetStatTypeStr(StatType::kGroupId)); in CreateStatMetadata()
48 builder.GetOrCreateStatMetadata(GetStatTypeStr(StatType::kStepName)); in CreateStatMetadata()
49 builder.GetOrCreateStatMetadata(GetStatTypeStr(StatType::kIsEager)); in CreateStatMetadata()
50 builder.GetOrCreateStatMetadata(GetStatTypeStr(StatType::kSelectedGroupIds)); in CreateStatMetadata()
Dxplane_utils_test.cc188 stats_owner->AddStatValue(*plane->GetOrCreateStatMetadata(stats_name), in CreateXStats()
/external/tensorflow/tensorflow/core/profiler/convert/
Dxplane_to_trace_events_test.cc38 event1.AddStatValue(*host_plane.GetOrCreateStatMetadata("tf_op"), in CreateXSpace()
39 *host_plane.GetOrCreateStatMetadata("Relu")); in CreateXSpace()
46 event2.AddStatValue(*host_plane.GetOrCreateStatMetadata("tf_op"), in CreateXSpace()
47 *host_plane.GetOrCreateStatMetadata("Conv2D")); in CreateXSpace()
58 event3.AddStatValue(*device_plane.GetOrCreateStatMetadata("correlation id"), in CreateXSpace()
Dxplane_to_profile_response_test.cc42 event1.AddStatValue(*host_plane.GetOrCreateStatMetadata("tf_op"), in CreateXSpace()
43 *host_plane.GetOrCreateStatMetadata("Relu")); in CreateXSpace()
50 event2.AddStatValue(*host_plane.GetOrCreateStatMetadata("tf_op"), in CreateXSpace()
51 *host_plane.GetOrCreateStatMetadata("Conv2D")); in CreateXSpace()
61 event3.AddStatValue(*device_plane.GetOrCreateStatMetadata("correlation id"), in CreateXSpace()
Dxplane_to_op_stats_test.cc55 device_plane.AddStatValue(*device_plane.GetOrCreateStatMetadata("clock_rate"), in TEST()
57 device_plane.AddStatValue(*device_plane.GetOrCreateStatMetadata("core_count"), in TEST()
60 *device_plane.GetOrCreateStatMetadata("memory_bandwidth"), in TEST()
63 *device_plane.GetOrCreateStatMetadata("compute_cap_major"), in TEST()
66 *device_plane.GetOrCreateStatMetadata("compute_cap_minor"), in TEST()
Dop_stats_to_tf_stats_test.cc46 *plane->GetOrCreateStatMetadata("level 0"), in AddTensorFlowOpEvent()
47 *plane->GetOrCreateStatMetadata(std::move(tf_op_fullname))); in AddTensorFlowOpEvent()
62 event.ParseAndAddStatValue(*plane->GetOrCreateStatMetadata("kernel_details"), in AddTensorFlowOpEventWithKernelDetails()
Dxplane_to_op_metrics_db_test.cc44 *plane->GetOrCreateStatMetadata("level 0"), in AddTensorFlowOpEvent()
45 *plane->GetOrCreateStatMetadata(std::move(tf_op_fullname))); in AddTensorFlowOpEvent()
/external/tensorflow/tensorflow/core/profiler/internal/cpu/
Dmetadata_collector.cc83 *xplane.GetOrCreateStatMetadata(kHloProto); in CollectData()
Dhost_tracer_utils.cc76 xplane.GetOrCreateStatMetadata(metadata.key); in ConvertCompleteEventsToXPlane()