Searched defs:make_mask (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/nn/utils/ |
| H A D | prune.py | 618 def make_mask(t, dim, nchannels, nchannels_toprune): function
735 def make_mask(t, dim, indices): function
|
| /aosp_15_r20/external/skia/gm/ |
| H A D | imagemasksubset.cpp | 35 sk_sp<SkImage> make_mask(const sk_sp<SkSurface>& surface) { in make_mask() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| H A D | array_ops_test.py | 148 def CheckVersusNumpy(self, ndims_mask, arr_shape, make_mask=None, axis=None): argument
151 make_mask = lambda shape: self.rng.randint(0, 2, size=shape).astype(bool) function
227 make_mask = lambda shape: np.zeros(shape, dtype=bool) function
|