Searched refs:IsSingleGpuGraph (Results 1 – 3 of 3) sorted by relevance
102 TEST(IsSingleGpuGraph, ReturnsTrue) { in TEST() argument111 EXPECT_TRUE(IsSingleGpuGraph(*root.graph())); in TEST()114 TEST(IsSingleGpuGraph, ReturnsFalseForCpuGraph) { in TEST() argument123 EXPECT_FALSE(IsSingleGpuGraph(*root.graph())); in TEST()126 TEST(IsSingleGpuGraph, ReturnsFalseForMultiGpuGraph) { in TEST() argument135 EXPECT_FALSE(IsSingleGpuGraph(*root.graph())); in TEST()
84 bool IsSingleGpuGraph(const Graph& g);
263 bool IsSingleGpuGraph(const Graph& g) { in IsSingleGpuGraph() function299 IsSingleGpuGraph(**options.graph) ? xla_global_jit_level.single_gpu in GetGlobalJitLevelForGraph()