Searched refs:ndims_mask (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | array_ops_test.py | 121 def CheckVersusNumpy(self, ndims_mask, arr_shape, make_mask=None, axis=None): argument
126 mask = make_mask(arr_shape[:ndims_mask])
128 mask = make_mask(arr_shape[axis:ndims_mask + axis])
148 ndims_mask = 1
150 self.CheckVersusNumpy(ndims_mask, arr_shape, axis=1)
154 ndims_mask = 2
156 self.CheckVersusNumpy(ndims_mask, arr_shape, axis=1)
160 ndims_mask = 1
162 self.CheckVersusNumpy(ndims_mask, arr_shape)
166 ndims_mask = 1
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 1328 ndims_mask = shape_mask.ndims
1330 if ndims_mask == 0:
1332 if ndims_mask is None:
1337 shape_tensor[axis:axis + ndims_mask].assert_is_compatible_with(shape_mask)
1339 leading_size = gen_math_ops.prod(shape(tensor)[axis:axis + ndims_mask], [0])
1343 shape(tensor)[axis + ndims_mask:]
1345 first_dim = shape_tensor[axis:axis + ndims_mask].num_elements()
1348 .concatenate(shape_tensor[axis + ndims_mask:]))
|