Home
last modified time | relevance | path

Searched defs:mul_op (Results 1 – 21 of 21) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/
H A 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
H A 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
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/cutlass_extensions/gemm/warp/
H A Dmma_tensorop_dequantizer.h272 multiplies<ExpandedMmaOperandB> mul_op; in dequantize() local
364 multiplies<FragmentDequantizedOperand> mul_op; in dequantize() local
455 multiplies<MmaOperandB> mul_op; in dequantize() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/tests/
H A Dfuse_binary_into_following_affine_test.cc84 auto* mul_op = new MulOperator; in TEST_F() local
126 auto* mul_op = new MulOperator; in TEST_F() local
H A Didentify_l2_normalization_test.cc84 auto mul_op = new MulOperator; in RunIdentifyL2Normalization() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Didentify_hardswish.cc50 const auto* mul_op = GetOpWithInput(*model, add_with_relu6_op->outputs[0]); in Run() local
H A Didentify_prelu.cc67 const auto* mul_op = GetOpWithOutput(*model, add_op->inputs[1]); in Run() local
H A Dresolve_multiply_by_zero.cc59 auto* mul_op = mul_it->get(); in Run() local
H A Dresolve_batch_normalization.cc63 auto* mul_op = new MulOperator; in Run() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
H A Dsquared_difference.cc71 auto* mul_op = graph_builder_->AddNode(GetTFLiteNodeID()); in PopulateSubGraph() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/
H A Dreplace_cast_hacks_with_tf_xla_ops.cc90 TF::MulOp mul_op = builder.create<TF::MulOp>(loc, zp, reduced); in CalculateZeroPointOffset() local
/aosp_15_r20/external/mesa3d/src/gallium/drivers/lima/ir/gp/
H A Dcodegen.h156 gpir_codegen_mul_op mul_op : 3; member
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/
H A Ddevice_transform_patterns.cc277 auto mul_op = in matchAndRewrite() local
/aosp_15_r20/external/mesa3d/src/gallium/drivers/vc4/
H A Dvc4_qpu_schedule.c327 uint32_t mul_op = QPU_GET_FIELD(inst, QPU_OP_MUL); in calculate_deps() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/
H A Dexport_tensorflow.cc712 tensorflow::NodeDef* mul_op = tensorflow_graph->add_node(); in ConvertMulOperator() local
941 tensorflow::NodeDef* mul_op = tensorflow_graph->add_node(); in ConvertL2NormalizationOperator() local
H A Dimport_tensorflow.cc1817 auto* mul_op = new MulOperator; in ConvertBatchNormWithGlobalNormalizationOperator() local
1879 auto* mul_op = new MulOperator; in ConvertFusedBatchNormOperator() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
H A Dprepare_tf.cc802 auto mul_op = rewriter.create<TF::MulOp>( in matchAndRewrite() local
H A Doptimize.cc736 auto mul_op = dyn_cast_or_null<TFL::MulOp>(fc_op.input().getDefiningOp()); in matchAndRewrite() local
/aosp_15_r20/external/mesa3d/src/compiler/nir/
H A Dnir_loop_analyze.c963 nir_op mul_op; in test_iterations() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
H A Dlegalize_tf.cc460 auto mul_op = CreateOpAndInfer<tosa::MulOp>(rewriter, op->getLoc(), in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dlegalize_tf.cc3062 auto mul_op = rewriter.create<MulOp>(loc, tanh_op, half); in matchAndRewrite() local