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()
2474 if (!IsPack(*node) || NumNonControlInputs(*node) != 1 || in SimplifyPack()3146 if (ctx == nullptr || !has_fetch_ || NumNonControlInputs(parent) != 2) { in PrepareConstantPushDown()3509 if (!IsAnyMul(*node) || NumNonControlInputs(*node) != 2) return false; in MulConvPushDown()3654 const int num_non_control_inputs = NumNonControlInputs(*node); in PartialAssocOpConstFolding()3731 const int num_non_control_inputs = NumNonControlInputs(*node); in PartialConcatConstFolding()3854 const int num_regular_inputs = NumNonControlInputs(*node); in MergeConcat()3877 const int n_parent_inputs = NumNonControlInputs(*parent); in MergeConcat()3918 (*parent->mutable_attr())["N"].set_i(NumNonControlInputs(*parent) - 1); in MergeConcat()
771 return IsAggregate(*node) && NumNonControlInputs(*node) > 1 && in IsSupported()2905 return IsPack(*node) && NumNonControlInputs(*node) > 1 && in IsSupported()2913 while (IsPack(*input) && NumNonControlInputs(*node) > 1 && in TrySimplify()
364 EXPECT_EQ(NumNonControlInputs(*add_node), 2); in TEST_F()401 EXPECT_EQ(NumNonControlInputs(*add_node), 2); in TEST_F()
803 if (NumNonControlInputs(node) == 1 && IsAggregate(node)) { in IsValueAndOrderAndShapePreserving()823 if (NumNonControlInputs(node) == 1 && IsAggregate(node)) { in IsValueAndOrderPreserving()
361 int NumNonControlInputs(const NodeDef& node);
192 int NumNonControlInputs(const NodeDef& node) { in NumNonControlInputs() function