Searched refs:ndims_mask (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | array_ops_test.py | 131 def CheckVersusNumpy(self, ndims_mask, arr_shape, make_mask=None, axis=None): argument
136 mask = make_mask(arr_shape[:ndims_mask])
138 mask = make_mask(arr_shape[axis:ndims_mask + axis])
158 ndims_mask = 1
161 self.CheckVersusNumpy(ndims_mask, arr_shape, axis=1)
165 ndims_mask = 2
168 self.CheckVersusNumpy(ndims_mask, arr_shape, axis=1)
172 ndims_mask = 1
175 self.CheckVersusNumpy(ndims_mask, arr_shape)
179 ndims_mask = 1
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 1830 ndims_mask = shape_mask.ndims
1832 if ndims_mask == 0:
1834 if ndims_mask is None:
1842 shape_tensor[axis:axis + ndims_mask].assert_is_compatible_with(shape_mask)
1844 leading_size = gen_math_ops.prod(shape(tensor)[axis:axis + ndims_mask], [0])
1849 shape(tensor)[axis + ndims_mask:]
1856 first_dim = shape_tensor[axis:axis + ndims_mask].num_elements()
1859 [first_dim]).concatenate(shape_tensor[axis + ndims_mask:]))
|