Home
last modified time | relevance | path

Searched refs:mul_op (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Didentify_prelu.cc67 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()
128 DeleteArrayIfUsedOnce(mul_op->inputs[1], model); in Run()
Dresolve_multiply_by_zero.cc59 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()
158 DeleteArrayIfUsedOnce(mul_op->inputs[0], model); in Run()
159 DeleteArrayIfUsedOnce(mul_op->inputs[1], model); in Run()
Dresolve_batch_normalization.cc63 auto* mul_op = new MulOperator; in Run() local
71 mul_op->inputs = {bn_op->inputs[0], mul_param_name}; in Run()
72 mul_op->outputs = {mul_name}; in Run()
75 AddMessageF("Splitting %s into %s and %s", LogName(*bn_op), LogName(*mul_op), in Run()
79 auto& intermediate_array = model->GetOrCreateArray(mul_op->outputs[0]); in Run()
84 auto mul_it = model->operators.emplace(add_it, mul_op); in Run()
86 DCHECK_EQ(mul_it->get(), mul_op); in Run()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/tests/
Dfuse_binary_into_following_affine_test.cc82 auto* mul_op = new MulOperator; in TEST_F() local
83 mul_op->inputs = {"Input", "MulInput2"}; in TEST_F()
84 mul_op->outputs = {"MulOutput"}; in TEST_F()
85 model_->operators.push_back(std::unique_ptr<Operator>(mul_op)); in TEST_F()
124 auto* mul_op = new MulOperator; in TEST_F() local
125 mul_op->inputs = {"Input", "MulInput2"}; in TEST_F()
126 mul_op->outputs = {"MulOutput"}; in TEST_F()
127 model_->operators.push_back(std::unique_ptr<Operator>(mul_op)); in TEST_F()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dfold_batch_norms_test.cc64 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
Dquantize_nodes_test.cc176 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/tensorflow/tensorflow/contrib/quantize/python/
Dquantize_graph_test.py309 mul_op = graph.get_operation_by_name('test/Mul')
311 mul_op, graph,
313 mul_op = graph.get_operation_by_name('test/Mul_1')
315 mul_op, graph,
/external/mesa3d/src/compiler/nir/
Dnir_loop_analyze.c408 nir_op mul_op; in test_iterations() local
413 mul_op = nir_op_fmul; in test_iterations()
419 mul_op = nir_op_imul; in test_iterations()
431 nir_eval_const_opcode(mul_op, 1, bit_size, mul_src); in test_iterations()
/external/tensorflow/tensorflow/lite/toco/
Dexport_tensorflow.cc702 tensorflow::NodeDef* mul_op = tensorflow_graph->add_node(); in ConvertMulOperator() local
703 mul_op->set_op("Mul"); in ConvertMulOperator()
704 mul_op->set_name(src_op.outputs[0]); in ConvertMulOperator()
706 *mul_op->add_input() = src_op.inputs[0]; in ConvertMulOperator()
707 *mul_op->add_input() = src_op.inputs[1]; in ConvertMulOperator()
708 (*mul_op->mutable_attr())["T"].set_type( in ConvertMulOperator()
928 tensorflow::NodeDef* mul_op = tensorflow_graph->add_node(); in ConvertL2NormalizationOperator() local
929 mul_op->set_op("Mul"); in ConvertL2NormalizationOperator()
930 mul_op->set_name(src_op.outputs[0]); in ConvertL2NormalizationOperator()
931 *mul_op->add_input() = src_op.inputs[0]; in ConvertL2NormalizationOperator()
[all …]
Dimport_tensorflow.cc1689 auto* mul_op = new MulOperator; in ConvertBatchNormWithGlobalNormalizationOperator() local
1690 mul_op->inputs.push_back(rsqrt); in ConvertBatchNormWithGlobalNormalizationOperator()
1691 mul_op->inputs.push_back(node.input(4)); in ConvertBatchNormWithGlobalNormalizationOperator()
1692 mul_op->outputs.push_back(multiplier); in ConvertBatchNormWithGlobalNormalizationOperator()
1693 model->operators.emplace_back(mul_op); in ConvertBatchNormWithGlobalNormalizationOperator()
1749 auto* mul_op = new MulOperator; in ConvertFusedBatchNormOperator() local
1750 mul_op->inputs.push_back(rsqrt_op_name); in ConvertFusedBatchNormOperator()
1751 mul_op->inputs.push_back(gamma_input); in ConvertFusedBatchNormOperator()
1752 mul_op->outputs.push_back(multiplier); in ConvertFusedBatchNormOperator()
1753 model->operators.emplace_back(mul_op); in ConvertFusedBatchNormOperator()
/external/mesa3d/src/gallium/drivers/vc4/
Dvc4_qpu_schedule.c353 uint32_t mul_op = QPU_GET_FIELD(inst, QPU_OP_MUL); in calculate_deps() local
375 if (mul_op != QPU_M_NOP) { in calculate_deps()
/external/kernel-headers/original/uapi/asm-mips/asm/
Dinst.h67 madd_op, maddu_op, mul_op, spec2_3_unused_op, enumerator
/external/tensorflow/tensorflow/python/kernel_tests/
Dcontrol_flow_ops_py_test.py185 mul_op = math_ops.multiply(enter_data, enter_five)
186 exit_op = control_flow_ops.exit(mul_op)
270 mul_op = math_ops.multiply(switch_op[1], five)
271 merge_op = control_flow_ops.merge([add_op, mul_op])[0]