Searched refs:ColocateWith (Results 1 – 5 of 5) sorted by relevance
166 TEST(CCOpTest, ColocateWith) { in TEST() argument169 auto c2 = Const(root.WithOpName("c2").ColocateWith(c1), 2); in TEST()174 auto c3 = Const(root.WithOpName("c3").ColocateWith(c2), 3); in TEST()179 auto c4 = Const(root.WithOpName("c4").ColocateWith(a), 5); in TEST()183 auto c5 = Const(root.WithOpName("c5").ColocateWith(c3).ColocateWith(c4), 6); in TEST()188 Scope with_colocate = root.ColocateWith(c3).ColocateWith(c4); in TEST()
155 Scope ColocateWith(const Operation& op) const;157 Scope ColocateWith(const Output& out) const { return ColocateWith(out.op()); } in ColocateWith() function
461 Scope Scope::ColocateWith(const Operation& op) const { in ColocateWith() function in tensorflow::Scope
1612 Output c = ops::Const(s.WithOpName("c").ColocateWith(a), 3.0f, {1}); in TEST_F()
1282 TEST_F(CApiColocationTest, ColocateWith) { in TEST_F() argument