Searched refs:TfOp (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/core/profiler/utils/ |
| D | tf_op_utils_test.cc | 27 TfOp tf_op = ParseTfOpFullname(kName); in TEST()
36 TfOp tf_op = ParseTfOpFullname(kName); in TEST()
45 TfOp tf_op = ParseTfOpFullname(kName); in TEST()
54 TfOp tf_op = ParseTfOpFullname(kName); in TEST()
63 TfOp tf_op = ParseTfOpFullname(kName); in TEST()
72 TfOp tf_op = ParseTfOpFullname(kName); in TEST()
82 TfOp tf_op = ParseTfOpFullname(kName); in TEST()
91 TfOp tf_op = ParseTfOpFullname(kName); in TEST()
101 TfOp tf_op = ParseTfOpFullname(kName); in TEST()
110 TfOp tf_op = ParseTfOpFullname(kName); in TEST()
[all …]
|
| D | tf_op_utils.h | 45 struct TfOp { struct
50 TfOp ParseTfOpFullname(absl::string_view tf_op_fullname); argument
53 std::vector<absl::string_view> ParseTfNameScopes(const TfOp& tf_op);
57 std::string TfOpEventName(const TfOp& tf_op);
70 inline bool IsDatasetOp(const TfOp& tf_op) { in IsDatasetOp()
|
| D | tf_op_utils.cc | 67 TfOp ParseTfOpFullname(absl::string_view tf_op_fullname) { in ParseTfOpFullname()
69 TfOp tf_op = {Category::kUnknown, tf_op_fullname, kUnknownOp}; in ParseTfOpFullname()
98 std::vector<absl::string_view> ParseTfNameScopes(const TfOp& tf_op) { in ParseTfNameScopes()
106 std::string TfOpEventName(const TfOp& tf_op) { in TfOpEventName()
|
| D | cost_utils.cc | 89 TfOp tf_op; in Predict()
|
| D | op_metrics_db_utils.cc | 118 TfOp tf_op = ParseTfOpFullname(device_op_metrics.provenance()); in CreateTfMetricsDbFromDeviceOpMetricsDb()
|
| D | derived_timeline.cc | 74 TfOp tf_op = ParseTfOpFullname(tf_op_full_name); in ProcessTfOpEvent()
|
| /external/tensorflow/tensorflow/core/profiler/convert/ |
| D | xplane_to_op_metrics_db.cc | 60 TfOp tf_op;
141 const absl::flat_hash_map<int64, TfOp>& tf_ops, in CollectTfActivities()
147 const TfOp* tf_op = gtl::FindOrNull(tf_ops, event.Id()); in CollectTfActivities()
166 absl::flat_hash_map<int64, TfOp> CollectTfOpsFromHostThreadsXPlane( in CollectTfOpsFromHostThreadsXPlane()
168 absl::flat_hash_map<int64, TfOp> tf_ops; in CollectTfOpsFromHostThreadsXPlane()
175 TfOp tf_op = ParseTfOpFullname(metadata.name()); in CollectTfOpsFromHostThreadsXPlane()
184 const XLineVisitor& line, const absl::flat_hash_map<int64, TfOp>& tf_ops) { in ConvertHostThreadsXLineToTfMetricsDbData()
201 absl::flat_hash_map<int64, TfOp> tf_ops = in ConvertHostThreadsXPlaneToOpMetricsDb()
239 TfOp tf_op = ParseTfOpFullname(tf_op_full_name); in ConvertDeviceTraceXPlaneToOpMetricsDb()
|
| D | xplane_to_op_metrics_db.h | 43 absl::flat_hash_map<int64, TfOp> CollectTfOpsFromHostThreadsXPlane(
47 const XLineVisitor& line, const absl::flat_hash_map<int64, TfOp>& tf_ops);
|
| D | xplane_to_kernel_stats_db.cc | 77 tensorflow::profiler::TfOp tf_op = ParseTfOpFullname(tf_op_fullname); in ConvertDeviceTraceXPlaneToKernelReports()
|
| D | xplane_to_op_stats.cc | 115 absl::flat_hash_map<int64, TfOp> tf_ops = in ProcessHostPlane()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 606 template <typename BinaryOp, typename TfOp>
636 rewriter.replaceOpWithNewOp<TfOp>(reduce_op, reduce_op.getType(0), input, in matchAndRewrite()
1086 template <typename BinaryOp, typename TfOp>
1152 rewriter.replaceOpWithNewOp<TfOp>(scatter_op, in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_patterns.td | 641 foreach TfOp = [TF_ExpandDimsOp, TF_ReshapeOp, TF_SqueezeOp, ] in {
642 def : Pat<(TfOp:$res $arg, $ignored),
|