Home
last modified time | relevance | path

Searched refs:IsMul (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/
Dop_types.h106 bool IsMul(const NodeDef& node);
Dop_types.cc327 bool IsMul(const NodeDef& node) { return node.op() == "Mul"; } in IsMul() function
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer.cc785 if ((!IsMul(*input) && !IsAnyDiv(*input)) || (IsMul(*input) && has_div) || in GetCommonFactors()
806 } else if (IsMul(*input)) { in GetCommonFactors()
915 return IsMul(node) || IsAdd(node); in IsBinaryAssociative()
2105 TF_RETURN_IF_TRUE(!IsMul(*source)); in TrySimplify()
2320 return IsMul(*node) && node->input(0) == node->input(1); in IsSupported()
Dconstant_folding.cc2641 const bool is_mul = IsMul(*node) || IsLogicalAnd(*node); in SimplifyArithmeticOperations()
2829 if (has_fetch_ && (IsAdd(*node) || IsMul(*node)) && in ConstantPushDown()
2905 if (!IsMul(*node) || NumNonControlInputs(*node) != 2) return false; in MulConvPushDown()
Dlayout_optimizer.cc321 IsMinimum(node) || IsMod(node) || IsMul(node) || IsPolygamma(node) || in IsBinaryOp()