Searched refs:NumNonControlOutputs (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/ |
| D | utils_test.cc | 258 return node.op() == "Neg" && NumNonControlOutputs(node, node_map) == 1; in TEST_F()
322 TEST_F(UtilsTest, NumNonControlOutputs) { in TEST_F() argument
358 EXPECT_EQ(NumNonControlOutputs(*add_node, node_map), 2); in TEST_F()
|
| D | utils.h | 259 int NumNonControlOutputs(const NodeDef& node, const NodeMap& node_map);
|
| D | utils.cc | 274 int NumNonControlOutputs(const NodeDef& node, const NodeMap& node_map) { in NumNonControlOutputs() function
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 139 IsValuePreserving(node) && NumNonControlOutputs(node, node_map) == 1; in GetTailOfValuePreservingChain()
150 IsIdempotent(node) && NumNonControlOutputs(node, node_map) == 1; in GetTailOfIdempotentChain()
943 if (NumNonControlOutputs(node, *ctx().node_map) != 1) { in IsAbsorbableByOptimizedNodesGroup()
1359 NumNonControlOutputs(*input, *ctx().node_map) > 1) { in TrySimplify()
1437 if (NumNonControlOutputs(*node, *ctx().node_map) > num_split) { in IsSupported()
1768 (NumNonControlOutputs(*y, *ctx().node_map) == 1)) { in TrySimplify()
1798 (NumNonControlOutputs(*b, *ctx().node_map) == 1)) { in TrySimplify()
1825 (NumNonControlOutputs(*x, *ctx().node_map) == 1)) { in TrySimplify()
2106 TF_RETURN_IF_TRUE(NumNonControlOutputs(*source, *ctx().node_map) != 1); in TrySimplify()
|
| D | dependency_optimizer.cc | 121 if (NumNonControlOutputs(node, *node_map_) > 0) { in SafeToConvertToNoOp()
|
| D | constant_folding.cc | 2390 NumNonControlOutputs(*fanout, *node_map_) > 0) { in SimplifySwitch()
2853 NumNonControlOutputs(*op_child_node, *node_map_) > 1) { in ConstantPushDown()
2928 NumNonControlOutputs(*conv_node, *node_map_) > 1 || in MulConvPushDown()
|