| /external/tensorflow/tensorflow/core/ir/ |
| D | tf_op_names.cc | 25 StringAttr op_name = op->getName().getIdentifier(); in IsAdd() local
33 StringAttr op_name = op->getName().getIdentifier(); in IsAddN() local
38 StringAttr op_name = op->getName().getIdentifier(); in IsAll() local
43 StringAttr op_name = op->getName().getIdentifier(); in IsAngle() local
48 StringAttr op_name = op->getName().getIdentifier(); in IsAny() local
53 StringAttr op_name = op->getName().getIdentifier(); in IsAnyDiv() local
59 StringAttr op_name = op->getName().getIdentifier(); in IsAnyBatchMatMul() local
64 StringAttr op_name = op->getName().getIdentifier(); in IsAnyMatMul() local
70 StringAttr op_name = op->getName().getIdentifier(); in IsAnyMax() local
76 StringAttr op_name = op->getName().getIdentifier(); in IsAnyMaxPool() local
[all …]
|
| D | tf_op_registry.cc | 30 StringRef op_name) { in IsStatefulImpl()
44 StringRef op_name = op->getName().stripDialect(); in isStateful() local
|
| /external/tensorflow/tensorflow/core/kernels/mlir_generated/ |
| D | base_unary_ops_test.h | 40 void SetOpKernel(const std::string& op_name, const TensorShape& shape, in SetOpKernel()
58 void RunAndExpectResult(const std::string& op_name, const TensorShape& shape, in RunAndExpectResult()
101 void TestImpl(const std::string& op_name, const TensorShape& shape, in TestImpl()
119 void Test(const std::string& op_name, const TensorShape& shape, in Test()
130 void Test(const std::string& op_name, const TensorShape& shape, in Test()
141 void Test(const std::string& op_name, const TensorShape& shape, in Test()
150 void TestEmptyShape(const std::string& op_name, in TestEmptyShape()
181 #define GENERATE_DEFAULT_TEST(op_name, InT, OutT, baseline_callback, config) \ argument
185 #define GENERATE_DEFAULT_TEST_2(op_name, InT, BaselineT, OutT, BaselineOutT, \ argument
192 op_name, InT, OutT, input_values, baseline_callback, config) \ argument
[all …]
|
| D | base_binary_ops_test.h | 44 void SetOpKernel(const std::string& op_name, const TensorShape& lhs_shape, in SetOpKernel()
68 void RunAndExpectResult(const std::string& op_name, in RunAndExpectResult()
95 void RunAndExpectInvalidArgument(const std::string& op_name, in RunAndExpectInvalidArgument()
111 void TestIncompatibleShapes(const std::string& op_name, in TestIncompatibleShapes()
129 void TestEqualShapes(const std::string& op_name, const TensorShape& shape, in TestEqualShapes()
162 void TestOneScalar(const std::string& op_name, T scalar_input, in TestOneScalar()
194 void TestOneEffectiveScalar(const std::string& op_name, T scalar_input, in TestOneEffectiveScalar()
230 void TestBroadcastingExpand(const std::string& op_name, in TestBroadcastingExpand()
259 void TestBroadcastingInDim(const std::string& op_name, in TestBroadcastingInDim()
288 void TestBroadcasting(const std::string& op_name, in TestBroadcasting()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/tools/ |
| D | tflite_op_coverage_spec_getters_gen.cc | 70 auto op_name = op.getCppClassName(); in EmitDynamicRangeOp() local
95 for (const auto &op_name : weight_only) { in EmitDynamicRangeOp() local
189 std::string op_name = op.getCppClassName().str(); in GenerateStaticQuantOp() local
214 for (const auto &op_name : result) { in EmitStaticInt8PerAxisQuantOp() local
233 for (const auto &op_name : result) { in EmitStaticInt8PerTensorQuantOp() local
252 for (const auto &op_name : result) { in EmitStaticUInt8PerAxisQuantOp() local
271 for (const auto &op_name : result) { in EmitStaticUInt8PerTensorQuantOp() local
303 for (const auto &op_name : result) { in EmitStaticQuantWithInt16ActOp() local
|
| /external/tensorflow/tensorflow/core/profiler/lib/ |
| D | traceme_encode.h | 136 absl::string_view op_name, absl::string_view op_type) { in TraceMeOp()
140 TF_ATTRIBUTE_ALWAYS_INLINE inline std::string TraceMeOp(const char* op_name, in TraceMeOp()
146 std::string&& op_name, absl::string_view op_type) { in TraceMeOp()
153 absl::string_view op_name, absl::string_view op_type) { in TraceMeOpOverride()
158 const char* op_name, const char* op_type) { in TraceMeOpOverride()
|
| D | scoped_memory_debug_annotation.h | 47 explicit ScopedMemoryDebugAnnotation(const char* op_name) { in ScopedMemoryDebugAnnotation()
55 explicit ScopedMemoryDebugAnnotation(const char* op_name, int64_t step_id) { in ScopedMemoryDebugAnnotation()
67 const char* op_name, const char* region_type, int32_t data_type, in ScopedMemoryDebugAnnotation()
81 const char* op_name, int64_t step_id, const char* region_type, in ScopedMemoryDebugAnnotation()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | placement_utils.cc | 33 static bool IsPinnableOp(StringPiece op_name) { in IsPinnableOp()
62 bool IsColocationExempt(StringPiece op_name) { in IsColocationExempt()
67 bool IsFunction(StringPiece op_name) { in IsFunction()
81 bool* result, StringPiece op_name, in MaybePinSmallOpsToCpu()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | dataset_stateful_op_allowlist.h | 28 Status Add(string op_name) { in Add()
33 Status Remove(string op_name) { in Remove()
38 bool Contains(const string& op_name) { return op_names_.count(op_name); } in Contains()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_gradient_exclusions.cc | 32 const char *op_name; member
52 const tensorflow::string &op_name) { in OpGradientUnusedInputIndices()
429 const tensorflow::string &op_name) { in OpGradientUnusedOutputIndices()
|
| D | execute.py | 27 def quick_execute(op_name, num_outputs, inputs, attrs, ctx, name=None): argument
71 def execute_with_cancellation(op_name, argument
125 def execute_with_callbacks(op_name, num_outputs, inputs, attrs, ctx, name=None): argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/python_tests/ |
| D | tf_math_ops_test.py | 43 def mlir_func_1d(op_name): argument
51 def test_1d(op_name, fn, vectorize=False, lb=-1.0, ub=1.0, rtol_enum=Rtol.BASE): argument
95 def test_op(self, op_name, fn, vectorize, rtol_enum): argument
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_def_builder.cc | 102 Node* SourceOp(const string& op_name, const GraphDefBuilder::Options& opts) { in SourceOp()
109 Node* UnaryOp(const string& op_name, NodeOut input, in UnaryOp()
118 Node* BinaryOp(const string& op_name, NodeOut a, NodeOut b, in BinaryOp()
127 Node* TernaryOp(const string& op_name, NodeOut a, NodeOut b, NodeOut c, in TernaryOp()
|
| D | mkl_graph_util.h | 195 static inline bool IsMklOp(const string& op_name, DataType T, in IsMklOp()
255 static inline bool IsMklQuantizedOp(const string& op_name, DataType Tinput, in IsMklQuantizedOp()
267 static inline bool IsMklOp(const string& op_name, DataType T) { in IsMklOp()
283 static inline bool IsMklElementWiseOp(const string& op_name, DataType T) { in IsMklElementWiseOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | converter_gen.cc | 75 static inline std::string GetOperatorBuilderName(StringRef op_name) { in GetOperatorBuilderName()
85 static inline bool IsLstmOp(const StringRef op_name) { in IsLstmOp()
99 StringRef op_name = def->getName().drop_front(4); // Strip 'TFL_' prefix in EmitOptionBuilders() local
175 StringRef op_name = def->getName().drop_front(4); in EmitOperatorBuilders() local
243 StringRef op_name = def->getName().drop_front(4); in EmitGetBuiltinOpCode() local
272 auto op_name = def->getValueAsString("opName"); in EmitOperandNumbers() local
324 StringRef op_name = def->getName().drop_front(4); in EmitBuildOperator() local
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | allowlisted_flex_ops.cc | 813 bool IsAllowedTFTextOpForFlex(const std::string& op_name) { in IsAllowedTFTextOpForFlex()
835 bool IsAllowedSentencePieceOpForFlex(const std::string& op_name) { in IsAllowedSentencePieceOpForFlex()
|
| /external/tensorflow/tensorflow/core/kernels/batching_util/ |
| D | batch_resource_base.cc | 46 int32_t execution_batch_size, const string& op_name) { in RecordPaddingSize()
59 int32_t execution_batch_size, const string& op_name) { in RecordPaddingSizeV2()
74 const string& op_name) { in RecordInputBatchSize()
86 const string& op_name) { in RecordInputBatchSizeV2()
100 const string& op_name) { in RecordBatchSize()
111 const string& op_name) { in RecordProcessedBatchSize()
124 const string& op_name) { in RecordProcessedBatchSizeV2()
136 const string& op_name, int32_t batch_size) { in RecordBatchDelayUs()
149 const string& op_name, int32_t batch_size) { in RecordBatchDelayUsV2()
164 const string& op_name) { in RecordBatchParamBatchTimeoutMicros()
[all …]
|
| /external/tensorflow/tensorflow/lite/ |
| D | model_xnnpack_test.cc | 52 const std::string op_name = GetOpNameByRegistration(first_node_reg); in TEST() local
80 const std::string op_name = GetOpNameByRegistration(first_node_reg); in TEST() local
|
| D | tensorflow_profiler_logger_shim.cc | 28 TFLITE_ATTRIBUTE_WEAK void OnTfLiteOpPrepare(const char* op_name, in OnTfLiteOpPrepare()
47 const char* op_name, int subgraph_index, int node_index) { in OnTfLiteOpInvoke()
|
| /external/tensorflow/tensorflow/lite/tools/serialization/ |
| D | option_writer_generator.cc | 233 for (const auto& op_name : ops_) { in BuildOpToOptionMap() local
338 void GenerateImportForOp(FILE* fp, const std::string& op_name, in GenerateImportForOp()
467 for (const auto& op_name : option->ops()) { in GenerateImport() local
478 for (const auto& op_name : option->ops()) { in GenerateImport() local
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | op_compatibility_lib.cc | 42 static string FileNameFromOpName(const string& op_name) { in FileNameFromOpName()
129 const string& op_name = op_list_.op(cur).name(); in ValidateCompatible() local
235 const string& op_name = op_list_.op(cur).name(); in ValidateCompatible() local
|
| /external/tensorflow/tensorflow/core/tfrt/eager/ |
| D | op_cache.cc | 26 string_view op_name, OpHandler* op_handler, string_view device_name, in GetOrAddOp()
59 Expected<CoreRuntimeOp*> OpCache::GetOrAddXlaOp(string_view op_name, in GetOrAddXlaOp()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | python_api_info_test.py | 40 def makeConverterForGenOp(self, op_name): argument
176 def testInitializeFromRegisteredOp(self, op_name, debug_info): argument
|
| /external/tensorflow/tensorflow/core/profiler/utils/ |
| D | tf_op_utils.cc | 51 absl::string_view op_name = name_scopes_and_op_name.back(); in DeriveOpType() local
73 bool IsTfOpName(absl::string_view op_name) { in IsTfOpName()
89 bool IsJaxOpNameAndType(absl::string_view op_name, absl::string_view op_type) { in IsJaxOpNameAndType()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | unary_ops_composition.cc | 99 for (absl::string_view op_name : op_names_) { in UnaryOpsCompositionOp() local
109 for (absl::string_view op_name : op_names_) { in Compile() local
|