Searched refs:NumNonControlInputs (Results 1 – 8 of 8) sorted by relevance
69 return IsAddN(node) && NumNonControlInputs(node) <= 1; in IsTrivialOp()385 const int num_non_control_inputs = NumNonControlInputs(*node); in SplitIdentityNInputs()
2294 if (IsPack(*node) && NumNonControlInputs(*node) == 1 && in SimplifyPack()2830 NumNonControlInputs(*node) == 2) { in ConstantPushDown()2905 if (!IsMul(*node) || NumNonControlInputs(*node) != 2) return false; in MulConvPushDown()3014 NumNonControlInputs(*node) > 0) { in PartialConstPropThroughIdentityN()3079 const int num_non_control_inputs = NumNonControlInputs(*node); in PartialAssocOpConstFolding()3155 const int num_non_control_inputs = NumNonControlInputs(*node); in PartialConcatConstFolding()3277 const int index = NumNonControlInputs(*node) - 1; in MergeConcat()3304 (*parent->mutable_attr())["N"].set_i(NumNonControlInputs(*parent) - 1); in MergeConcat()
683 return IsAggregate(*node) && NumNonControlInputs(*node) > 1 && in IsSupported()2375 return IsAggregate(*node) && NumNonControlInputs(*node) > 0 && in IsSupported()
256 int NumNonControlInputs(const NodeDef& node);
704 if (NumNonControlInputs(node) == 1 && IsAggregate(node)) { in IsValueAndOrderAndShapePreserving()724 if (NumNonControlInputs(node) == 1 && IsAggregate(node)) { in IsValueAndOrderPreserving()
264 int NumNonControlInputs(const NodeDef& node) { in NumNonControlInputs() function
356 EXPECT_EQ(NumNonControlInputs(*add_node), 2); in TEST_F()
2085 if (NumNonControlInputs(node) == 0) { in InferStatically()