Home
last modified time | relevance | path

Searched refs:bn_node (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler_test.cc2654 const NodeState* bn_node = nullptr; in TEST_F() local
2660 bn_node = &node_state; in TEST_F()
2666 CHECK_NOTNULL(bn_node); in TEST_F()
2670 ValidateNodeDefs({"z4"}, bn_node->outputs.at(-1)); in TEST_F()
2671 ValidateNodeDefs({"z1"}, bn_node->outputs.at(0)); in TEST_F()
2673 ValidateNodeDefs({"z2", "z3", "z2", "z3"}, bn_node->outputs.at(2)); in TEST_F()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_transposer_test.cc795 auto* bn_node = context.graph_view->GetNode("bn"); in TEST_F() local
796 ASSERT_NE(bn_node, nullptr); in TEST_F()
797 ASSERT_EQ(bn_node->NumRegularFanins(), 5); in TEST_F()
798 VerifyRegularFaninMatch(bn_node, 0, input_transpose_node->GetName(), 0); in TEST_F()
799 VerifyRegularFaninMatch(bn_node, 1, "scale", 0); in TEST_F()
800 VerifyRegularFaninMatch(bn_node, 2, "offset", 0); in TEST_F()
801 VerifyRegularFaninMatch(bn_node, 3, "mean", 0); in TEST_F()
802 VerifyRegularFaninMatch(bn_node, 4, "var", 0); in TEST_F()
803 VerifyDataFormatAttributeMatch(bn_node, kDstFormat); in TEST_F()
809 VerifyRegularFaninMatch(output_transpose_node, 0, bn_node->GetName(), 0); in TEST_F()
[all …]