Home
last modified time | relevance | path

Searched defs:compute_mask (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/pytorch/torch/nn/utils/
H A Dprune.py34 def compute_mask(self, t, default_mask): member in BasePruningMethod
312 def compute_mask(self, t, default_mask): member in PruningContainer
413 def compute_mask(self, t, default_mask): member in Identity
452 def compute_mask(self, t, default_mask): member in RandomUnstructured
509 def compute_mask(self, t, default_mask): member in L1Unstructured
578 def compute_mask(self, t, default_mask): member in RandomStructured
686 def compute_mask(self, t, default_mask): member in LnStructured
798 def compute_mask(self, t, default_mask): member in CustomFromMask
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/layers/
H A Dmerge.py200 def compute_mask(self, inputs, mask=None): member in _Merge
539 def compute_mask(self, inputs, mask=None): member in Concatenate
725 def compute_mask(self, inputs, mask=None): member in Dot
H A Dembeddings.py158 def compute_mask(self, inputs, mask=None): member in Embedding
H A Ddense_attention.py180 def compute_mask(self, inputs, mask=None): member in BaseDenseAttention
H A Dcore.py116 def compute_mask(self, inputs, mask=None): member in Masking
964 def compute_mask(self, inputs, mask=None): member in Lambda
H A Dpooling.py986 def compute_mask(self, inputs, mask=None): member in GlobalAveragePooling1D
H A Drecurrent.py520 def compute_mask(self, inputs, mask): member in RNN
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/engine/
H A Dsequential.py408 def compute_mask(self, inputs, mask): member in Sequential
H A Dfunctional.py392 def compute_mask(self, inputs, mask): member in Functional
H A Dbase_layer_v1.py620 def compute_mask(self, inputs, mask=None): # pylint: disable=unused-argument member in Layer
H A Dbase_layer.py911 def compute_mask(self, inputs, mask=None): # pylint: disable=unused-argument member in Layer
/aosp_15_r20/external/mesa3d/src/broadcom/vulkan/
H A Dv3dv_cmd_buffer.c3327 const VkPipelineStageFlags2 compute_mask = in handle_barrier() local