| /external/tensorflow/tensorflow/compiler/mlir/tfr/utils/ |
| D | utils.cc | 71 std::string GetComposeFuncName(StringRef tf_op_name) { in GetComposeFuncName() argument
73 for (int i = 0; i < tf_op_name.size(); ++i) { in GetComposeFuncName()
74 if (tf_op_name[i] == '_') { in GetComposeFuncName()
79 if (tf_op_name[i] == '.') { in GetComposeFuncName()
81 } else if (tf_op_name[i] >= 'A' && tf_op_name[i] <= 'Z') { in GetComposeFuncName()
83 compose_func_name.push_back(tf_op_name[i] + 'a' - 'A'); in GetComposeFuncName()
85 compose_func_name.push_back(tf_op_name[i]); in GetComposeFuncName()
92 std::string tf_op_name; in GetTFOpName() local
101 tf_op_name.push_back(compose_func_name[i] + 'A' - 'a'); in GetTFOpName()
109 tf_op_name.push_back('.'); in GetTFOpName()
[all …]
|
| D | utils.h | 34 std::string GetComposeFuncName(StringRef tf_op_name);
|
| /external/tensorflow/tensorflow/core/profiler/utils/ |
| D | kernel_stats_utils.cc | 123 bool IsOpTensorCoreEligible(absl::string_view tf_op_name) { in IsOpTensorCoreEligible() argument
128 || absl::EndsWith(tf_op_name, "Conv2D") in IsOpTensorCoreEligible()
129 || absl::EndsWith(tf_op_name, "Conv2DBackpropFilter") in IsOpTensorCoreEligible()
130 || absl::EndsWith(tf_op_name, "Conv2DBackpropInput") in IsOpTensorCoreEligible()
131 || absl::EndsWith(tf_op_name, "Conv3D") in IsOpTensorCoreEligible()
132 || absl::EndsWith(tf_op_name, "DepthwiseConv2dNative") in IsOpTensorCoreEligible()
133 || absl::EndsWith(tf_op_name, "DepthwiseConv2dNativeBackpropFilter") in IsOpTensorCoreEligible()
134 || absl::EndsWith(tf_op_name, "DepthwiseConv2dNativeBackpropInput") in IsOpTensorCoreEligible()
136 || absl::StrContains(tf_op_name, "BatchMatMul") in IsOpTensorCoreEligible()
138 || absl::EndsWith(tf_op_name, "/MatMul") in IsOpTensorCoreEligible()
[all …]
|
| D | op_metrics_db_utils.cc | 42 absl::string_view tf_op_name, absl::string_view tf_op_type, in UpdateTfOpMetricsWithDeviceOpMetrics() argument
45 /*hlo_module_id=*/0, tf_op_name); in UpdateTfOpMetricsWithDeviceOpMetrics()
|
| D | derived_timeline.cc | 225 auto tf_op_name = symbol_resolver(hlo_module_name, hlo_op_names.back()); in DeriveEventsFromAnnotations() local
226 if (!tf_op_name.empty()) { in DeriveEventsFromAnnotations()
227 ProcessTfOpEvent(tf_op_name, offset_ps, duration_ps, group_id, &plane, in DeriveEventsFromAnnotations()
|
| D | kernel_stats_utils.h | 37 bool IsOpTensorCoreEligible(absl::string_view tf_op_name);
|
| /external/tensorflow/tensorflow/lite/tools/ |
| D | list_flex_ops_no_kernel.cc | 51 std::string tf_op_name = in AddFlexOpsFromModel() local
54 flex_ops->insert({tf_op_name, ""}); in AddFlexOpsFromModel()
|
| D | list_flex_ops.cc | 98 std::string tf_op_name = in AddFlexOpsFromModel() local
122 flex_ops->insert({tf_op_name, kernel_class}); in AddFlexOpsFromModel()
|
| /external/tensorflow/tensorflow/core/profiler/convert/ |
| D | xplane_to_memory_profile.cc | 204 alloc_meta->tf_op_name()); in UpdateDeallocation()
259 a_meta->tf_op_name(), a_meta->region_type(), in operator ()()
263 b_meta->tf_op_name(), b_meta->region_type(), in operator ()()
314 a_meta->tf_op_name() == b_meta->tf_op_name() && in operator ==()
|
| D | xplane_to_memory_profile_test.cc | 111 EXPECT_EQ(allocator_memory_profile.special_allocations().at(1).tf_op_name(), in TEST()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/passes/ |
| D | raise_to_tf.cc | 366 std::string tf_op_name = GetTFOpName(call_op.callee()); in CreateAndReplaceOp() local
367 OperationState new_state(loc, tf_op_name, inputs, output_types, attr_list); in CreateAndReplaceOp()
|
| /external/tensorflow/tensorflow/core/profiler/protobuf/ |
| D | memory_profile.proto | 50 string tf_op_name = 5; field
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | convolutional.py | 232 tf_op_name = self.__class__.__name__
233 if tf_op_name == 'Conv1D':
234 tf_op_name = 'conv1d' # Backwards compat.
242 name=tf_op_name)
|