Searched refs:xla_fbody (Results 1 – 2 of 2) sorted by relevance
304 std::unique_ptr<FunctionBody> xla_fbody; in TEST_F() local306 AttrSlice(), &fld, &xla_fbody)); in TEST_F()307 auto node_name_index = xla_fbody->graph->BuildNodeNameIndex(); in TEST_F()548 std::unique_ptr<FunctionBody> xla_fbody; in TEST_F() local550 AttrSlice(), &fld, &xla_fbody)); in TEST_F()551 Graph *xla_graph = xla_fbody->graph; in TEST_F()819 std::unique_ptr<FunctionBody> xla_fbody; in TEST_F() local821 AttrSlice(), &fld, &xla_fbody)); in TEST_F()822 Graph *xla_graph = xla_fbody->graph; in TEST_F()889 std::unique_ptr<FunctionBody> xla_fbody; in TEST_F() local[all …]
2900 std::unique_ptr<FunctionBody> xla_fbody; in ProcessHeadTailOutsideCompilation() local2903 AttrSlice(&iter.second.func_name_attrs.attr()), fld, &xla_fbody)); in ProcessHeadTailOutsideCompilation()2904 Graph* xla_graph = xla_fbody->graph; in ProcessHeadTailOutsideCompilation()