Searched defs:op_metrics (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/profiler/convert/ |
| H A D | op_profile_builder.cc | 46 void PopulateSymbolNode(const OpMetrics& op_metrics, Node* node) { in PopulateSymbolNode()
148 void PopulateOpMetricsNode(const OpMetrics& op_metrics, in PopulateOpMetricsNode()
185 void InsertFusedInstructions(const OpMetrics& op_metrics, Node* node) { in InsertFusedInstructions()
206 Node* OpProfileBuilder::AddOpNode(const OpMetrics& op_metrics, in AddOpNode()
221 Node* OpProfileBuilder::LookupOrAddDeduplicatedNode(const OpMetrics& op_metrics, in LookupOrAddDeduplicatedNode()
232 const OpMetrics& op_metrics, Program* program) { in LookupOrAddCategoryNode()
250 const OpMetrics& op_metrics) { in LookupOrAddProgramNode()
260 void OpProfileBuilder::AddOp(const OpMetrics& op_metrics) { in AddOp()
|
| H A D | xplane_to_op_metrics_db_test.cc | 107 OpMetricsDb op_metrics = ConvertHostThreadsXPlaneToOpMetricsDb(*xplane); in TEST() local
172 OpMetricsDb op_metrics = ConvertDeviceTraceXPlaneToOpMetricsDb(*xplane); in TEST() local
220 OpMetricsDb op_metrics = ConvertTpuDeviceTraceXPlaneToOpMetricsDb(*xplane); in TEST() local
|
| H A D | xplane_to_op_metrics_db.cc | 184 const XEventMetadataVisitor& hlo_event_metadata, OpMetrics* op_metrics) { in SetOpMetadataFromHloEventMetadata()
220 OpMetrics* op_metrics) { in SetOpMetricsFromHloEvent()
324 OpMetrics& op_metrics = op_metric_by_symbol[key.symbol_id.value()]; in ConvertTpuDeviceTraceXPlaneToOpMetricsDb() local
|
| H A D | op_stats_to_input_pipeline_analysis.cc | 330 for (const OpMetrics* op_metrics : SortedOpMetricsDb(all_op_metrics)) { in SelectInputOpMetrics() local
339 InputOpDetails ConvertOpMetricsToInputOpDetails(const OpMetrics& op_metrics, in ConvertOpMetricsToInputOpDetails()
482 for (const OpMetrics* op_metrics : input_op_metrics.input_op_metrics) { in GenerateHostResult() local
|
| H A D | op_stats_to_op_profile.cc | 54 for (const OpMetrics& op_metrics : metrics_db.metrics_db()) { in BuildOpProfileNodeTree() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/profiler/utils/ |
| H A D | op_utils.cc | 48 OpMetrics* op_metrics = LookupOrInsertNewOpMetrics(/*hlo_module_id=*/0, name); in EnterOp() local
85 OpMetrics* op_metrics = LookupOrInsertNewOpMetrics(program_id, name); in EnterOp() local
|
| H A D | op_metrics_db_utils.cc | 76 OpMetrics*& op_metrics = op_metrics_map_[hlo_module_id][name]; in LookupOrInsertNewOpMetrics() local
|