Searched defs:_IotaInitializer (Results 1 - 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| H A D | partitioned_variables_test.py | 306 def _IotaInitializer(shape, dtype=dtypes.float32, partition_info=None): function
311 val = _IotaInitializer(shape[1:], dtype)
489 self.assertAllClose([0., 1., 2., 3.], _IotaInitializer([4]))
491 _IotaInitializer([4, 2]))
494 _IotaInitializer)
496 slice0 = _IotaInitializer([5, 5])
497 slice1 = _IotaInitializer([4, 5])
498 slice2 = _IotaInitializer([4, 5])
|
Completed in 71 milliseconds