Searched refs:bn_node (Results 1 – 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/core/grappler/costs/ |
D | virtual_scheduler_test.cc | 2654 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/ |
D | generic_layout_optimizer_transposer_test.cc | 795 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 …]
|