| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | squared_difference.cc | 71 auto* mul_op = graph_builder_->AddNode(GetTFLiteNodeID()); in PopulateSubGraph() local
72 mul_op->SetOpType(OP_QuantizedMul_8x8to8); in PopulateSubGraph()
73 mul_op->AddInput(sub_out); in PopulateSubGraph()
74 mul_op->AddInput(sub_out); in PopulateSubGraph()
75 mul_op->AddInput(sub_min); in PopulateSubGraph()
76 mul_op->AddInput(sub_max); in PopulateSubGraph()
77 mul_op->AddInput(sub_min); in PopulateSubGraph()
78 mul_op->AddInput(sub_max); in PopulateSubGraph()
79 mul_op->AddInput(TensorID(output_min_const->GetID(), 0)); in PopulateSubGraph()
80 mul_op->AddInput(TensorID(output_max_const->GetID(), 0)); in PopulateSubGraph()
[all …]
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | identify_hardswish.cc | 49 const auto* mul_op = GetOpWithInput(*model, add_with_relu6_op->outputs[0]); in Run() local
50 ops.push_back(mul_op); in Run()
52 if (mul_op->type == OperatorType::kFakeQuant) { in Run()
53 mul_op = GetOpWithInput(*model, mul_op->outputs[0]); in Run()
54 ops.push_back(mul_op); in Run()
56 if (!IsBinaryOp(mul_op, OperatorType::kMul)) { in Run()
60 const auto* output_op = GetOpWithInput(*model, mul_op->outputs[0]); in Run()
81 std::vector<std::string> mul_inputs = mul_op->inputs; in Run()
|
| D | identify_prelu.cc | 67 const auto* mul_op = GetOpWithOutput(*model, add_op->inputs[1]); in Run() local
68 if (mul_op == nullptr || mul_op->type != OperatorType::kMul || in Run()
69 mul_op->inputs.size() != 2 || in Run()
70 mul_op->fused_activation_function != FusedActivationFunctionType::kNone) { in Run()
74 const auto neg_alpha_tensor_name = mul_op->inputs[0]; in Run()
76 const auto* relu_neg_input_op = GetOpWithOutput(*model, mul_op->inputs[1]); in Run()
127 DeleteArrayIfUnusedOutsideOfOp(mul_op->inputs[1], mul_op, model); in Run()
|
| D | resolve_multiply_by_zero.cc | 59 auto* mul_op = mul_it->get(); in Run() local
60 if (mul_op->type != OperatorType::kMul) { in Run()
63 const auto& output_array_name = mul_op->outputs[0]; in Run()
84 IsConstantParameterArray(*model, mul_op->inputs[0]), in Run()
85 IsConstantParameterArray(*model, mul_op->inputs[1]), in Run()
102 model->GetArray(mul_op->inputs[index_of_constant_input]); in Run()
157 DeleteOpAndArrays(model, mul_op); in Run()
|
| D | resolve_batch_normalization.cc | 63 auto* mul_op = new MulOperator; in Run() local
73 mul_op->inputs = {bn_op->inputs[0], mul_param_name}; in Run()
74 mul_op->outputs = {mul_name}; in Run()
77 AddMessageF("Splitting %s into %s and %s", LogName(*bn_op), LogName(*mul_op), in Run()
81 auto& intermediate_array = model->GetOrCreateArray(mul_op->outputs[0]); in Run()
86 auto mul_it = model->operators.emplace(add_it, mul_op); in Run()
88 DCHECK_EQ(mul_it->get(), mul_op); in Run()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/tests/ |
| D | fuse_binary_into_following_affine_test.cc | 83 auto* mul_op = new MulOperator; in TEST_F() local
84 mul_op->inputs = {"Input", "MulInput2"}; in TEST_F()
85 mul_op->outputs = {"MulOutput"}; in TEST_F()
86 model_->operators.push_back(std::unique_ptr<Operator>(mul_op)); in TEST_F()
125 auto* mul_op = new MulOperator; in TEST_F() local
126 mul_op->inputs = {"Input", "MulInput2"}; in TEST_F()
127 mul_op->outputs = {"MulOutput"}; in TEST_F()
128 model_->operators.push_back(std::unique_ptr<Operator>(mul_op)); in TEST_F()
|
| D | identify_l2_normalization_test.cc | 84 auto mul_op = new MulOperator; in RunIdentifyL2Normalization() local
85 mul_op->inputs = {"input0", rsqrt_op->outputs[0]}; in RunIdentifyL2Normalization()
86 mul_op->outputs = {"Muloutput"}; in RunIdentifyL2Normalization()
89 model.operators.push_back(std::unique_ptr<Operator>(mul_op)); in RunIdentifyL2Normalization()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | optimize.cc | 438 LogicalResult matchAndRewrite(TFL::MulOp mul_op, in matchAndRewrite()
442 if (mul_op.lhs().getType() != mul_op.getType()) return failure(); in matchAndRewrite()
446 Value constant_val = mul_op.rhs(); in matchAndRewrite()
451 dyn_cast_or_null<TFL::FullyConnectedOp>(mul_op.lhs().getDefiningOp()); in matchAndRewrite()
481 rewriter.create<ConstantOp>(mul_op.getLoc(), new_type, new_cst); in matchAndRewrite()
488 rewriter.create<TF::MulOp>(mul_op.getLoc(), filter, new_const_val).z(); in matchAndRewrite()
492 rewriter.create<TF::MulOp>(mul_op.getLoc(), bias, constant_val).z(); in matchAndRewrite()
496 FusedLoc::get({fc_op.getLoc(), mul_op.getLoc()}, fc_op.getContext()), in matchAndRewrite()
497 mul_op.getType(), in matchAndRewrite()
502 rewriter.getStringAttr(mul_op.fused_activation_function()), in matchAndRewrite()
[all …]
|
| D | prepare_tf.cc | 881 auto mul_op = rewriter.create<TF::MulOp>( in matchAndRewrite() local
883 rewriter.replaceOp(op, mul_op.getResult()); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/c/experimental/ops/ |
| D | math_ops.cc | 31 AbstractOperationPtr mul_op(ctx->CreateOperation()); in Mul() local
32 TF_RETURN_IF_ERROR(mul_op->Reset("Mul", /*raw_device_name=*/nullptr)); in Mul()
33 TF_RETURN_IF_ERROR(MaybeSetOpName(mul_op.get(), name)); in Mul()
34 TF_RETURN_IF_ERROR(mul_op->AddInput(inputs[0])); in Mul()
35 TF_RETURN_IF_ERROR(mul_op->AddInput(inputs[1])); in Mul()
37 return mul_op->Execute(outputs, &num_retvals); in Mul()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fold_batch_norms_test.cc | 64 Output mul_op = Mul(root.WithOpName("output"), conv_op, mul_values_op); in TestFoldBatchNormsConv2D() local
115 Output mul_op = Mul(root.WithOpName("output"), conv_op, mul_values_op); in TestFoldBatchNormsDepthwiseConv2dNative() local
166 Output mul_op = Mul(root.WithOpName("output"), conv_op, mul_values_op); in TestFoldBatchNormsConv2DShared() local
223 Output mul_op = Mul(root.WithOpName("output"), matmul_op, mul_values_op); in TestFoldBatchNormsMatMul() local
|
| D | quantize_nodes_test.cc | 176 Output mul_op = Mul(root.WithOpName("mul"), c_op, d_op); in TestIgnoreOps() local
1249 Output mul_op = Mul(root.WithOpName("mul_op"), add_op, c_op); in TestMergeDuplicateConsts() local
1298 Output mul_op = Mul(root.WithOpName("mul_op"), add_op, c_op); in TestMergeDuplicatesNested() local
|
| /external/mesa3d/src/gallium/drivers/lima/ir/gp/ |
| D | disasm.c | 252 switch (instr->mul_op) { in print_mul()
267 if (instr->mul_op == gpir_codegen_mul_op_complex2) in print_mul()
350 printf("\tunknown%u.m01 ", instr->mul_op); in print_mul()
|
| D | codegen.h | 156 gpir_codegen_mul_op mul_op : 3; member
|
| D | codegen.c | 126 code->mul_op = gpir_codegen_mul_op_complex1; in gpir_codegen_mul0_slot()
132 code->mul_op = gpir_codegen_mul_op_complex2; in gpir_codegen_mul0_slot()
138 code->mul_op = gpir_codegen_mul_op_select; in gpir_codegen_mul0_slot()
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | BUILD | 37 "mul_op.cc",
|
| /external/mesa3d/src/compiler/nir/ |
| D | nir_loop_analyze.c | 702 nir_op mul_op; in test_iterations() local
707 mul_op = nir_op_fmul; in test_iterations()
713 mul_op = nir_op_imul; in test_iterations()
724 eval_const_binop(mul_op, bit_size, iter_src, step, execution_mode); in test_iterations()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 707 tensorflow::NodeDef* mul_op = tensorflow_graph->add_node(); in ConvertMulOperator() local
708 mul_op->set_op("Mul"); in ConvertMulOperator()
709 mul_op->set_name(src_op.outputs[0]); in ConvertMulOperator()
711 *mul_op->add_input() = src_op.inputs[0]; in ConvertMulOperator()
712 *mul_op->add_input() = src_op.inputs[1]; in ConvertMulOperator()
713 (*mul_op->mutable_attr())["T"].set_type( in ConvertMulOperator()
936 tensorflow::NodeDef* mul_op = tensorflow_graph->add_node(); in ConvertL2NormalizationOperator() local
937 mul_op->set_op("Mul"); in ConvertL2NormalizationOperator()
938 mul_op->set_name(src_op.outputs[0]); in ConvertL2NormalizationOperator()
939 *mul_op->add_input() = src_op.inputs[0]; in ConvertL2NormalizationOperator()
[all …]
|
| D | import_tensorflow.cc | 1810 auto* mul_op = new MulOperator; in ConvertBatchNormWithGlobalNormalizationOperator() local
1811 mul_op->inputs.push_back(rsqrt); in ConvertBatchNormWithGlobalNormalizationOperator()
1812 mul_op->inputs.push_back(node.input(4)); in ConvertBatchNormWithGlobalNormalizationOperator()
1813 mul_op->outputs.push_back(multiplier); in ConvertBatchNormWithGlobalNormalizationOperator()
1814 model->operators.emplace_back(mul_op); in ConvertBatchNormWithGlobalNormalizationOperator()
1872 auto* mul_op = new MulOperator; in ConvertFusedBatchNormOperator() local
1873 mul_op->inputs.push_back(rsqrt_op_name); in ConvertFusedBatchNormOperator()
1874 mul_op->inputs.push_back(gamma_input); in ConvertFusedBatchNormOperator()
1875 mul_op->outputs.push_back(multiplier); in ConvertFusedBatchNormOperator()
1876 model->operators.emplace_back(mul_op); in ConvertFusedBatchNormOperator()
|
| /external/mesa3d/src/gallium/drivers/vc4/ |
| D | vc4_qpu_schedule.c | 327 uint32_t mul_op = QPU_GET_FIELD(inst, QPU_OP_MUL); in calculate_deps() local
349 if (mul_op != QPU_M_NOP) { in calculate_deps()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tfl.cc | 807 auto mul_op = in matchAndRewrite() local
813 rewriter, op, mul_op.getResult(), fused_activation_fn); in matchAndRewrite()
821 rewriter.replaceOp(op, {mul_op.getResult()}); in matchAndRewrite()
|
| D | legalize_tf.cc | 455 auto mul_op = rewriter.create<tosa::MulOp>( in matchAndRewrite() local
457 rewriter.replaceOp(op, {mul_op.getResult()}); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | control_flow_ops_py_test.py | 211 mul_op = math_ops.multiply(enter_data, enter_five)
212 exit_op = control_flow_ops.exit(mul_op)
296 mul_op = math_ops.multiply(switch_op[1], five)
297 merge_op = control_flow_ops.merge([add_op, mul_op])[0]
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 2663 auto mul_op = rewriter.create<MulOp>(loc, tanh_op, half); in matchAndRewrite() local
2664 auto add_op = rewriter.create<AddOp>(loc, mul_op, half); in matchAndRewrite()
|