Lines Matching refs:Assign

40 using tensorflow::ops::Assign;
107   auto assign_c = Assign(root.WithOpName("assign_c"), c, Mul(root, a, b));  in TEST()
125   auto restore = Assign(root.WithOpName("restore_a"), a, input);  in TEST()
126   auto double_a = Assign(root.WithOpName("double_a"), a,  in TEST()
151   auto restore = Assign(root.WithOpName("restore_a"), a, input);  in TEST()
152   auto double_a = Assign(root.WithOpName("double_a"), a,  in TEST()
184   auto restore = Assign(root.WithOpName("restore"), c,  in TEST()
242   auto init_a = Assign(root.WithOpName("init_a"), a,  in TEST()
245       Assign(root.WithOpName("init_b"), b, Const<int32_t>(root, {{2}}));  in TEST()
246   auto mul_a = Assign(root.WithOpName("mul_a"), a, Mul(root, a, b));  in TEST()
247   auto inc_b = Assign(root.WithOpName("inc_b"), b,  in TEST()
255   auto restore = Assign(root.WithOpName("restore"), a,  in TEST()