Home
last modified time | relevance | path

Searched refs:single_slice_dim (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dvariable_scope.py128 single_slice_dim = self.single_slice_dim(shape)
131 if single_slice_dim is None:
133 return self.var_offset[single_slice_dim]
135 def single_slice_dim(self, shape): member in _PartitionInfo
/external/tensorflow/tensorflow/python/kernel_tests/
Dvariable_scope_test.py1749 partition_info.single_slice_dim(None)
1753 partition_info.single_slice_dim([1, 2, 3])
1757 partition_info.single_slice_dim([6, 3])
1761 partition_info.single_slice_dim([1, 1])
1765 self.assertEqual(1, partition_info.single_slice_dim([9, 2]))
1768 self.assertEqual(0, partition_info.single_slice_dim([2, 3]))