Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/
Dutils_test.cc263 return node.op() == "Neg" && NumNonControlOutputs(node, node_map) == 1; in TEST_F()
327 TEST_F(UtilsTest, NumNonControlOutputs) { in TEST_F() argument
367 EXPECT_EQ(NumNonControlOutputs(*add_node, node_map), 2); in TEST_F()
Dutils.h367 int NumNonControlOutputs(const NodeDef& node, const NodeMap& node_map);
Dutils.cc259 int NumNonControlOutputs(const NodeDef& node, const NodeMap& node_map) { in NumNonControlOutputs() function
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer.cc144 IsValuePreserving(node) && NumNonControlOutputs(node, node_map) == 1; in GetTailOfValuePreservingChain()
155 IsIdempotent(node) && NumNonControlOutputs(node, node_map) == 1; in GetTailOfIdempotentChain()
1033 if (NumNonControlOutputs(node, *ctx().node_map) != 1) { in IsAbsorbableByOptimizedNodesGroup()
1463 NumNonControlOutputs(*input, *ctx().node_map) > 1) { in TrySimplify()
1541 if (NumNonControlOutputs(*node, *ctx().node_map) > num_split) { in IsSupported()
1898 (NumNonControlOutputs(*y, *ctx().node_map) == 1)) { in TrySimplify()
1934 (NumNonControlOutputs(*b, *ctx().node_map) == 1)) { in TrySimplify()
1965 (NumNonControlOutputs(*x, *ctx().node_map) == 1)) { in TrySimplify()
2022 NumNonControlOutputs(input, *ctx().node_map) > 1) || in TrySimplify()
2041 NumNonControlOutputs(*reshape_to_bypass, *ctx().node_map) > 1 || in TrySimplify()
[all …]
Dconstant_folding.cc3177 NumNonControlOutputs(*ctx->op_child, *node_map_) > 1) { in PrepareConstantPushDown()
3535 NumNonControlOutputs(*conv_node, *node_map_) > 1 || in MulConvPushDown()