Searched refs:modified_graph_def (Results 1 – 2 of 2) sorted by relevance
89 GraphDef modified_graph_def; in TestFlattenAtrousConv() local91 &modified_graph_def)); in TestFlattenAtrousConv()94 TF_ASSERT_OK(modified_session->Create(modified_graph_def)); in TestFlattenAtrousConv()98 EXPECT_EQ(3, modified_graph_def.node_size()); in TestFlattenAtrousConv()100 EXPECT_EQ("input_op", modified_graph_def.node(0).name()); in TestFlattenAtrousConv()101 EXPECT_EQ("weights_op", modified_graph_def.node(1).name()); in TestFlattenAtrousConv()102 EXPECT_EQ("output", modified_graph_def.node(2).name()); in TestFlattenAtrousConv()104 EXPECT_EQ("Const", modified_graph_def.node(0).op()); in TestFlattenAtrousConv()105 EXPECT_EQ("Const", modified_graph_def.node(1).op()); in TestFlattenAtrousConv()106 EXPECT_EQ(conv_op.node()->type_string(), modified_graph_def.node(2).op()); in TestFlattenAtrousConv()
698 GraphDef modified_graph_def = params.grappler_item->graph; in ConvertAfterShapes() local707 for (int i = 0; i < modified_graph_def.node_size(); i++) { in ConvertAfterShapes()708 NodeDef* node_def = modified_graph_def.mutable_node(i); in ConvertAfterShapes()709 TF_RETURN_IF_ERROR(MaybeRewriteCastToFp32(&modified_graph_def, node_def)); in ConvertAfterShapes()716 params.grappler_item->WithGraph(std::move(modified_graph_def)); in ConvertAfterShapes()