| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | op_factory.h | 26 OpBuilder* CreateArgMinMaxOpBuilder(GraphBuilder* graph_builder, int op_type);
27 OpBuilder* CreateActivationBuilder(GraphBuilder* graph_builder, int op_type);
28 OpBuilder* CreateArithmeticBuilder(GraphBuilder* graph_builder, int op_type);
30 int op_type);
31 OpBuilder* CreateConcatBuilder(GraphBuilder* graph_builder, int op_type);
32 OpBuilder* CreateConv2DBuilder(GraphBuilder* graph_builder, int op_type);
34 int op_type);
35 OpBuilder* CreatePool2DBuilder(GraphBuilder* graph_builder, int op_type);
36 OpBuilder* CreateReshapeBuilder(GraphBuilder* graph_builder, int op_type);
37 OpBuilder* CreateSoftmaxBuilder(GraphBuilder* graph_builder, int op_type);
[all …]
|
| D | op_builder.h | 41 int op_type; member
69 op_node_.op_type = hexagon_op_type; in OpBuilder()
80 void SetOpType(int op_type) { op_node_.op_type = op_type; } in SetOpType() argument
91 void SetConstNode() { op_node_.op_type = OP_Const; } in SetConstNode()
102 bool IsConstNode() const { return op_node_.op_type == OP_Const; } in IsConstNode()
126 int GetOpType() const { return op_node_.op_type; } in GetOpType()
235 OpBuilder* AddNodeFromTfLiteOp(int op_type, TfLiteNode* node,
252 OpBuilder* CreateOpBuilderFromTfLiteOp(int op_type, TfLiteNode* node);
297 graph_id_, op_node->node_id, op_node->op_type, op_node->padding_type, in Build()
302 op_node->op_type); in Build()
[all …]
|
| D | quantize_builder.h | 26 explicit QuantizeOpBuilder(GraphBuilder* graph_builder, int op_type) in QuantizeOpBuilder() argument
27 : OpBuilder(graph_builder, op_type) {} in QuantizeOpBuilder()
28 explicit QuantizeOpBuilder(GraphBuilder* graph_builder, int op_type, in QuantizeOpBuilder() argument
30 : OpBuilder(graph_builder, op_type) {} in QuantizeOpBuilder()
|
| D | activation_builder.h | 28 explicit ActivationOpBuilder(GraphBuilder* graph_builder, int op_type) in ActivationOpBuilder() argument
29 : OpBuilder(graph_builder, op_type) {} in ActivationOpBuilder()
30 explicit ActivationOpBuilder(GraphBuilder* graph_builder, int op_type, in ActivationOpBuilder() argument
32 : OpBuilder(graph_builder, op_type), relu_value_(relu_value) {} in ActivationOpBuilder()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/ |
| D | elementwise_test.cc | 41 OperationType op_type = OperationType::ABS; in TEST() local
43 SingleOpModel model({/*type=*/ToString(op_type), /*attributes=*/{}}, in TEST()
47 ASSERT_OK(model.Invoke(*NewElementwiseNodeShader(op_type))); in TEST()
53 OperationType op_type = OperationType::COS; in TEST() local
55 SingleOpModel model({/*type=*/ToString(op_type), /*attributes=*/{}}, in TEST()
59 ASSERT_OK(model.Invoke(*NewElementwiseNodeShader(op_type))); in TEST()
65 OperationType op_type = OperationType::COPY; in TEST() local
67 SingleOpModel model({/*type=*/ToString(op_type), /*attributes=*/{}}, in TEST()
71 ASSERT_OK(model.Invoke(*NewElementwiseNodeShader(op_type))); in TEST()
76 OperationType op_type = OperationType::ELU; in TEST() local
[all …]
|
| D | relu_test.cc | 46 OperationType op_type = OperationType::RELU; in TEST_F() local
50 SingleOpModel model({ToString(op_type), attr}, {GetTensorRef(0)}, in TEST_F()
59 OperationType op_type = OperationType::RELU; in TEST_F() local
63 SingleOpModel model({ToString(op_type), attr}, {GetTensorRef(0)}, in TEST_F()
72 OperationType op_type = OperationType::RELU; in TEST_F() local
76 SingleOpModel model({ToString(op_type), attr}, {GetTensorRef(0)}, in TEST_F()
85 OperationType op_type = OperationType::RELU; in TEST_F() local
89 SingleOpModel model({ToString(op_type), attr}, {GetTensorRef(0)}, in TEST_F()
|
| D | prelu_test.cc | 89 OperationType op_type = OperationType::PRELU; variable
103 SingleOpModel model({ToString(op_type), attr}, {input}, {output});
115 OperationType op_type = OperationType::PRELU; variable
129 SingleOpModel model({ToString(op_type), attr}, {input}, {output});
141 OperationType op_type = OperationType::PRELU; variable
155 SingleOpModel model({ToString(op_type), attr}, {input}, {output});
167 OperationType op_type = OperationType::PRELU; variable
181 SingleOpModel model({ToString(op_type), attr}, {input}, {output});
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | reduce.cc | 57 std::string MakeOp(OperationType op_type, const std::string& a, in MakeOp() argument
59 if (op_type == OperationType::REDUCE_SUM || op_type == OperationType::MEAN) { in MakeOp()
61 } else if (op_type == OperationType::REDUCE_PRODUCT) { in MakeOp()
63 } else if (op_type == OperationType::REDUCE_MAXIMUM) { in MakeOp()
65 } else if (op_type == OperationType::REDUCE_MINIMUM) { in MakeOp()
112 Reduce::Reduce(const std::map<Axis, int>& axis_to_reduce, OperationType op_type, in Reduce() argument
145 ordered_axis_to_reduce, op_type); in Reduce()
163 OperationType op_type) { in GetReduceKernelCode() argument
289 if (op_type == OperationType::REDUCE_SUM || op_type == OperationType::MEAN) { in GetReduceKernelCode()
291 } else if (op_type == OperationType::REDUCE_PRODUCT) { in GetReduceKernelCode()
[all …]
|
| D | elementwise.cc | 29 const OperationType& op_type, in GetOneInputCode() argument
33 switch (op_type) { in GetOneInputCode()
104 std::string GetTwoInputCode(const OperationType& op_type, in GetTwoInputCode() argument
110 switch (op_type) { in GetTwoInputCode()
185 const OperationDef& definition, const OperationType& op_type, in CreateElementwiseOneRuntimeOneScalar() argument
195 op.code_ += GetTwoInputCode(op_type, "in_out_value", "in_out_value", in CreateElementwiseOneRuntimeOneScalar()
204 const OperationType& op_type, in CreateElementwiseTwoInput() argument
226 result.code_ += GetTwoInputCode(op_type, "in_out_value", "in_out_value", in CreateElementwiseTwoInput()
235 const OperationType& op_type, in CreateElementwiseTwoInput() argument
260 result.code_ += GetTwoInputCode(op_type, "in_out_value", "in_out_value", in CreateElementwiseTwoInput()
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/vectorization/ |
| D | vectorizer_registry.h | 35 Vectorizer* Get(const string& op_type);
38 void Register(const string& op_type, std::unique_ptr<Vectorizer> vectorizer);
48 VectorizerRegistration(const string& op_type, in VectorizerRegistration() argument
50 VectorizerRegistry::Global()->Register(op_type, std::move(vectorizer)); in VectorizerRegistration()
56 #define REGISTER_VECTORIZER(op_type, vectorizer) \ argument
57 REGISTER_VECTORIZER_UNIQ_HELPER(__COUNTER__, op_type, vectorizer)
59 #define REGISTER_VECTORIZER_UNIQ_HELPER(ctr, op_type, vectorizer) \ argument
60 REGISTER_VECTORIZER_UNIQ(ctr, op_type, vectorizer)
62 #define REGISTER_VECTORIZER_UNIQ(ctr, op_type, vectorizer) \ argument
65 op_type, ::std::unique_ptr<::tensorflow::grappler::Vectorizer>( \
|
| D | vectorizer_registry.cc | 28 Vectorizer* VectorizerRegistry::Get(const string& op_type) { in Get() argument
29 auto found = vectorizers_.find(op_type); in Get()
36 void VectorizerRegistry::Register(const string& op_type, in Register() argument
38 auto existing = Get(op_type); in Register()
40 << "Vectorizer for op type: " << op_type << " already registered"; in Register()
42 op_type, std::move(vectorizer))); in Register()
|
| /external/tensorflow/tensorflow/python/debug/lib/ |
| D | dumping_callback.py | 61 def is_op_type_function(op_type): argument
62 return compat.as_bytes(op_type).startswith(_FUNCTION_PREFIXES)
292 op_type, argument
310 if op_type in ("Placeholder", "PlaceholderWithDefault"):
318 op_type != "Const"):
336 op_type, argument
387 if (not self._should_dump_tensor(op_type, tensor.dtype) or
407 op_type, tensor, debug_tensor, tensor_debug_mode))
421 if (not self._should_dump_tensor(op_type, tensor.dtype) or
434 op_type, tensor, debug_tensor, tensor_debug_mode))
[all …]
|
| D | debug_events_monitors.py | 89 op_type, argument
98 self._op_type = op_type
112 def op_type(self): member in InfNanAlert
155 op_type, argument
186 op_type,
199 op_type, argument
230 op_type,
239 op_type,
253 op_type,
273 tensor_value, execution.wall_time, execution.op_type, output_slot,
[all …]
|
| D | dumping_callback_test.py | 162 executed_op_types.append(execution.op_type)
164 if execution.op_type in ("AddV2", "Mul", "RealDiv"):
169 "Unexpected device name from eager op %s" % execution.op_type)
184 if execution.op_type in ("AddV2", "Mul", "RealDiv"):
189 if execution.op_type in ("AddV2", "Mul", "RealDiv"):
196 if execution.op_type in ("AddV2", "Mul", "RealDiv"):
207 if execution.op_type in ("AddV2", "Mul", "RealDiv"):
216 tensor_values[execution.op_type].append(
291 executed_op_types = [trace.op_type for trace in graph_exec_traces
292 if trace.op_type != "Const"]
[all …]
|
| D | debug_events_monitors_test.py | 88 self.assertEqual(execution.op_type, "MatMul")
139 self.assertStartsWith(execution.op_type, "__inference_unique_sum")
149 self.assertEqual(traces[0].op_type, "Placeholder")
151 self.assertEqual(traces[1].op_type, "Unique")
154 self.assertEqual(traces[2].op_type, "Sum")
162 self.assertEqual(traces[0].op_type, "Placeholder")
164 self.assertEqual(traces[1].op_type, "Unique")
167 self.assertEqual(traces[2].op_type, "Sum")
178 self.assertEqual(traces[0].op_type, "Placeholder")
184 self.assertEqual(traces[1].op_type, "Unique")
[all …]
|
| D | debug_events_reader.py | 372 op_type, argument
375 self._op_type = op_type
379 def op_type(self): member in ExecutionDigest
389 "op_type": self.op_type,
439 execution_digest.op_type,
541 (graph_op_creation_digest.op_name, graph_op_creation_digest.op_type))
664 op_type, argument
673 self._op_type = op_type
686 def op_type(self): member in GraphOpCreationDigest
721 "op_type": self.op_type,
[all …]
|
| /external/tensorflow/tensorflow/core/profiler/lib/ |
| D | traceme_encode.h | 133 absl::string_view op_name, absl::string_view op_type) { in TraceMeOp() argument
134 return absl::StrCat(op_name, ":", op_type); in TraceMeOp()
138 const char* op_type) { in TraceMeOp() argument
139 return absl::StrCat(op_name, ":", op_type); in TraceMeOp()
143 std::string&& op_name, absl::string_view op_type) { in TraceMeOp() argument
144 absl::StrAppend(&op_name, ":", op_type); in TraceMeOp()
150 absl::string_view op_name, absl::string_view op_type) { in TraceMeOpOverride() argument
151 return absl::StrCat("#tf_op=", op_name, ":", op_type, "#"); in TraceMeOpOverride()
155 const char* op_name, const char* op_type) { in TraceMeOpOverride() argument
156 return absl::StrCat("#tf_op=", op_name, ":", op_type, "#"); in TraceMeOpOverride()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | gradient_input_output_exclusions.py | 263 def _get_num_inputs_outputs(op_type): argument
286 op_def = op_def_registry.get(op_type)
310 for op_type in ops._gradient_registry.list(): # pylint: disable=protected-access
311 if op_type in _EXCLUDED_OPS:
313 num_values = _get_num_inputs_outputs(op_type)[0 if attr_name ==
315 gradient_fn = ops._gradient_registry.lookup(op_type) # pylint: disable=protected-access
319 entries[op_type] = "{\"%s\"}," % op_type
325 entries[op_type] = "{\"%s\"}," % op_type
331 entries[op_type] = "{\"%s\", %d, {%s}}," % (
332 op_type, len(unused_tensor_list), ", ".join(
[all …]
|
| D | tape.py | 181 def record_operation(op_type, output_tensors, input_tensors, backward_function, argument
184 pywrap_tfe.TFE_Py_TapeSetRecordOperation(op_type, output_tensors,
189 def record_operation_backprop_only(op_type, output_tensors, input_tensors, argument
192 pywrap_tfe.TFE_Py_TapeSetRecordOperationBackprop(op_type, output_tensors,
197 def record_operation_forwardprop_only(op_type, output_tensors, input_tensors, argument
215 op_type, output_tensors, input_tensors, backward_function,
|
| /external/XNNPACK/tools/ |
| D | generate-vunary-test.py | 33 op_type = {
50 return op_type, batch_tile, arch, isa
160 def generate_test_cases(ukernel, op_type, batch_tile, isa): argument
178 "VUnOpMicrokernelTester::OpType::%s" % op_type,
187 "OP_TYPE": op_type,
222 op_type, batch_tile, arch, isa = split_ukernel_name(name)
227 test_case = generate_test_cases(name, op_type, batch_tile, isa)
|
| D | generate-vbinary-test.py | 37 op_type = {
64 return op_type, activation_type, batch_tile, arch, isa
257 def generate_test_cases(ukernel, op_type, activation_type, tester, batch_tile, isa): argument
277 test_args.append("%s::OpType::%s" % (tester, op_type))
286 "OP_TYPE": op_type,
336 op_type, activation_type, batch_tile, arch, isa = split_ukernel_name(name)
341 test_case = generate_test_cases(name, op_type, activation_type,
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | image_grad_test.cc | 54 void MakeOp(const OpType op_type, const Tensor& x_data, const Input& y_shape, in MakeOp() argument
58 switch (op_type) { in MakeOp()
79 void TestResizedShapeForType(const OpType op_type, const bool align_corners, in TestResizedShapeForType() argument
84 MakeOp<T>(op_type, x_data, {4, 6}, align_corners, half_pixel_centers, &x, in TestResizedShapeForType()
94 void TestResizedShape(OpType op_type) { in TestResizedShape() argument
100 TestResizedShapeForType<Eigen::half>(op_type, align_corners, in TestResizedShape()
102 TestResizedShapeForType<float>(op_type, align_corners, in TestResizedShape()
104 TestResizedShapeForType<double>(op_type, align_corners, in TestResizedShape()
111 void TestResizeToSmallerAndAlign(const OpType op_type, in TestResizeToSmallerAndAlign() argument
117 MakeOp<X_T>(op_type, x_data, {2, 3}, align_corners, half_pixel_centers, &x, in TestResizeToSmallerAndAlign()
[all …]
|
| /external/llvm-project/llvm/test/CodeGen/NVPTX/ |
| D | ld-st-addrrspace.py | 56 for op_type, volatile, space in product(
73 "type": op_type,
78 "ptx_reg": llvm_type_to_ptx_reg[op_type],
79 "ptx_type": llvm_type_to_ptx_type[op_type],
|
| /external/tensorflow/tensorflow/core/profiler/utils/ |
| D | tf_op_utils.cc | 50 bool IsTfOpType(absl::string_view op_type) { in IsTfOpType() argument
52 return RE2::FullMatch(op_type, *kTfOpTypeRegEx); in IsTfOpType()
55 bool IsJaxOpType(absl::string_view op_type) { in IsJaxOpType() argument
57 return RE2::FullMatch(op_type, *kJaxOpTypeRegEx); in IsJaxOpType()
60 bool IsJaxOpNameAndType(absl::string_view op_name, absl::string_view op_type) { in IsJaxOpNameAndType() argument
61 if (op_name.empty() || !IsJaxOpType(op_type)) return false; in IsJaxOpNameAndType()
64 return absl::StrContains(split_result.back(), op_type); in IsJaxOpNameAndType()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/ |
| D | if_conditional.hlotxt | 8 %log.9 = f32[] log(%get-tuple-element.8), metadata={op_type="Log" op_name="cond/Log"}
15 …%exponential.14 = f32[] exponential(%get-tuple-element.13), metadata={op_type="Exp" op_name="cond/…
24 %constant.3 = f32[] constant(10), metadata={op_type="Less" op_name="Less"}
27 …%compare.4 = pred[] compare(%arg0.1, %constant.3), direction=LT, metadata={op_type="Less" op_name=…
30 %tuple.5 = (f32[]) tuple(%arg0.1), metadata={op_type="If" op_name="cond/Merge_if"}
45 …true_computation=%then_branch, false_computation=%else_branch, metadata={op_type="If" op_name="con…
48 …%get-tuple-element.17 = f32[] get-tuple-element(%conditional.16), index=0, metadata={op_type="If" …
|