HomeSort by relevance Sort by last modified time
    Searched refs:colocation_groups (Results 1 - 10 of 10) sorted by null

  /external/tensorflow/tensorflow/contrib/framework/python/ops/
variables_test.py 439 self.assertEqual(a.initial_value.op.colocation_groups(),
440 a.op.colocation_groups())
442 self.assertEqual(b.initial_value.op.colocation_groups(),
443 b.op.colocation_groups())
445 self.assertEqual(c.initial_value.op.colocation_groups(),
446 c.op.colocation_groups())
448 self.assertEqual(d.initial_value.op.colocation_groups(),
449 d.op.colocation_groups())
466 self.assertEqual(a.initial_value.op.colocation_groups(),
467 a.op.colocation_groups())
    [all...]
  /external/tensorflow/tensorflow/python/ops/
gradients_test.py 171 self.assertEqual(gw.op.colocation_groups(), wx.op.colocation_groups())
185 self.assertEqual(gw1.op.colocation_groups(), wx.op.colocation_groups())
188 self.assertTrue(wx.op.colocation_groups() != gw2.op.colocation_groups())
204 self.assertEqual(gw1.op.colocation_groups(), w.op.colocation_groups())
207 self.assertTrue(w.op.colocation_groups() != gw2.op.colocation_groups())
    [all...]
  /external/tensorflow/tensorflow/python/framework/
importer_test.py 710 self.assertEqual(a.colocation_groups(), [b"loc:@imported_graph/A"])
711 self.assertEqual(b.colocation_groups(), [b"loc:@imported_graph/A"])
728 self.assertEqual(a.colocation_groups(), [b"loc:@imported_graph/A"])
729 self.assertEqual(b.colocation_groups(), [b"loc:@imported_graph/A"])
744 self.assertEqual(a.colocation_groups(), [b"loc:@imported_graph/A"])
745 self.assertEqual(b.colocation_groups(), [b"loc:@imported_graph/A"])
774 self.assertEqual(a.colocation_groups(), [b"loc:@imported_graph/A"])
775 self.assertEqual(b.colocation_groups(), [b"loc:@imported_graph/B"])
776 self.assertEqual(c.colocation_groups(),
795 self.assertEqual(b.colocation_groups(), [b"loc:@A"]
    [all...]
ops_test.py     [all...]
ops.py 1741 def colocation_groups(self): member in class:Operation
    [all...]
  /external/tensorflow/tensorflow/python/kernel_tests/
variables_test.py 406 self.assertEqual([b"loc:@v"], assign_op.op.colocation_groups())
463 self.assertEqual(expected_group_v1, v1.op.colocation_groups())
465 self.assertEqual(expected_group_v1, i.op.colocation_groups())
469 self.assertEqual(expected_group_v2, v2.op.colocation_groups())
471 self.assertEqual(expected_group_v2, i.op.colocation_groups())
resource_variable_ops_test.py 225 self.assertEqual(v.handle.op.colocation_groups(),
226 v.initializer.inputs[1].op.colocation_groups())
control_flow_ops_py_test.py     [all...]
  /external/tensorflow/tensorflow/contrib/kfac/python/kernel_tests/
fisher_factors_test.py 51 self.assertEqual([b'loc:@a'], a.op.colocation_groups())
52 self.assertEqual([b'loc:@b'], b.op.colocation_groups())
60 self.assertEqual([b'loc:@a'], a.op.colocation_groups())
61 self.assertEqual([b'loc:@a'], b.op.colocation_groups())
  /external/tensorflow/tensorflow/python/training/
moving_averages_test.py 393 self.assertEqual([b"loc:@v1"], ema.average(v1).op.colocation_groups())

Completed in 2555 milliseconds