Searched refs:NumNonControlInputs (Results 1 – 7 of 7) sorted by relevance
78 return IsAddN(node) && NumNonControlInputs(node) <= 1; in IsTrivialOp()403 const int num_non_control_inputs = NumNonControlInputs(*node); in SplitIdentityNInputs()
2424 if (!IsPack(*node) || NumNonControlInputs(*node) != 1 || in SimplifyPack()3096 if (ctx == nullptr || !has_fetch_ || NumNonControlInputs(parent) != 2) { in PrepareConstantPushDown()3453 if (!IsAnyMul(*node) || NumNonControlInputs(*node) != 2) return false; in MulConvPushDown()3595 const int num_non_control_inputs = NumNonControlInputs(*node); in PartialAssocOpConstFolding()3672 const int num_non_control_inputs = NumNonControlInputs(*node); in PartialConcatConstFolding()3795 const int num_regular_inputs = NumNonControlInputs(*node); in MergeConcat()3818 const int n_parent_inputs = NumNonControlInputs(*parent); in MergeConcat()3859 (*parent->mutable_attr())["N"].set_i(NumNonControlInputs(*parent) - 1); in MergeConcat()
709 return IsAggregate(*node) && NumNonControlInputs(*node) > 1 && in IsSupported()
361 EXPECT_EQ(NumNonControlInputs(*add_node), 2); in TEST_F()398 EXPECT_EQ(NumNonControlInputs(*add_node), 2); in TEST_F()
795 if (NumNonControlInputs(node) == 1 && IsAggregate(node)) { in IsValueAndOrderAndShapePreserving()815 if (NumNonControlInputs(node) == 1 && IsAggregate(node)) { in IsValueAndOrderPreserving()
347 int NumNonControlInputs(const NodeDef& node);
207 int NumNonControlInputs(const NodeDef& node) { in NumNonControlInputs() function