Searched refs:NumControlInputs (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | scoped_allocator_optimizer_test.cc | 347 int NumControlInputs(NodeMap* node_map, const string& node_name) { in NumControlInputs() function in tensorflow::grappler::__anon3b02b9070111::ScopedAllocatorOptimizerTest
545 EXPECT_EQ(NumControlInputs(&node_map, sa_concat->name()), 2); in TEST_F()
552 EXPECT_EQ(NumControlInputs(&node_map, fused_abs->name()), 0); in TEST_F()
560 EXPECT_EQ(NumControlInputs(&node_map, "ctl3"), 1); in TEST_F()
561 EXPECT_EQ(NumControlInputs(&node_map, "ctl4"), 1); in TEST_F()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | utils_test.cc | 365 EXPECT_EQ(NumControlInputs(*add_node), 1); in TEST_F()
370 EXPECT_EQ(NumControlInputs(*mul_node), 0); in TEST_F()
402 EXPECT_EQ(NumControlInputs(*add_node), 1); in TEST_F()
403 EXPECT_EQ(NumControlInputs(*mul_node), 0); in TEST_F()
|
| D | utils.h | 358 int NumControlInputs(const NodeDef& node);
|
| D | utils.cc | 203 int NumControlInputs(const NodeDef& node) { in NumControlInputs() function
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | make_deterministic.cc | 300 int num_control_deps = NumControlInputs(*map_node); in SplitMap()
|