| /external/tensorflow/tensorflow/core/profiler/convert/ |
| D | xplane_to_op_metrics_db.cc | 141 const absl::flat_hash_map<int64, TfOp>& tf_ops, in CollectTfActivities() argument
145 line.ForEachEvent([&tf_ops, &tf_op_id, in CollectTfActivities()
147 const TfOp* tf_op = gtl::FindOrNull(tf_ops, event.Id()); in CollectTfActivities()
168 absl::flat_hash_map<int64, TfOp> tf_ops; in CollectTfOpsFromHostThreadsXPlane() local
177 tf_ops.try_emplace(metadata.id(), tf_op); in CollectTfOpsFromHostThreadsXPlane()
180 return tf_ops; in CollectTfOpsFromHostThreadsXPlane()
184 const XLineVisitor& line, const absl::flat_hash_map<int64, TfOp>& tf_ops) { in ConvertHostThreadsXLineToTfMetricsDbData() argument
186 if (!tf_ops.empty()) { in ConvertHostThreadsXLineToTfMetricsDbData()
188 CollectTfActivities(line, tf_ops, &tf_activities); in ConvertHostThreadsXLineToTfMetricsDbData()
201 absl::flat_hash_map<int64, TfOp> tf_ops = in ConvertHostThreadsXPlaneToOpMetricsDb() local
[all …]
|
| D | xplane_to_op_metrics_db.h | 47 const XLineVisitor& line, const absl::flat_hash_map<int64, TfOp>& tf_ops);
|
| D | xplane_to_op_stats.cc | 115 absl::flat_hash_map<int64, TfOp> tf_ops = in ProcessHostPlane() local
121 ConvertHostThreadsXLineToTfMetricsDbData(line, tf_ops), &combiner); in ProcessHostPlane()
|
| /external/tensorflow/tensorflow/core/profiler/utils/ |
| D | derived_timeline.cc | 148 DerivedXLineBuilder tf_ops(&plane, kThreadIdTfOp, kTensorFlowOpLineName, in DeriveEventsFromAnnotations() local
152 start_timestamp_ns, {&tf_ops}); in DeriveEventsFromAnnotations()
228 &tf_name_scope, &tf_ops); in DeriveEventsFromAnnotations()
232 &plane, &tf_name_scope, &tf_ops); in DeriveEventsFromAnnotations()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tf_data_optimization.td | 17 include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td"
|
| D | optimize.td | 17 include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td"
|
| D | canonicalize.td | 19 include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td"
|
| D | legalize_hlo_patterns.td | 20 include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td"
|
| D | lower_tf.td | 17 include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td"
|
| D | decompose_resource_ops.td | 17 include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ |
| D | BUILD | 34 "ir/tf_ops.td",
43 "ir/tf_ops.td",
81 "g3doc/tf_ops.md",
93 td_file = "ir/tf_ops.td",
136 td_file = "ir/tf_ops.td",
189 td_file = "ir/tf_ops.td",
210 td_file = "ir/tf_ops.td",
391 "ir/tf_ops.h",
435 "ir/tf_ops.h",
478 "ir/tf_ops.h",
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | tensorlist_patterns.td | 18 include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td"
|
| D | legalize_patterns.td | 21 include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td"
|
| D | optimize_patterns.td | 22 include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td"
|
| /external/tensorflow/tensorflow/compiler/mlir/tfjs/transforms/ |
| D | optimize_pattern.td | 21 include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td"
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | tf_legalize_patterns.td | 21 include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_helpers.inc | 17 // tf_ops.cc file. The helpers in here should be refactored and moved to
18 // tf_verifiers or tf_ops.
|
| D | tf_generated_ops.td | 21 // definition to `tf_ops.td` and perform the modification there.
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_patterns.td | 21 include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td"
|