Searched refs:DROPOUT_GEN_MASK_CNODE_INPUT_SIZE (Results 1 – 2 of 2) sorted by relevance
164 if (cnode_non_monad_size != DROPOUT_GEN_MASK_CNODE_INPUT_SIZE) { in GetDropoutGenMaskPrim()165 …ON) << "The size of dropout gen mask cnode's inputs must be " << DROPOUT_GEN_MASK_CNODE_INPUT_SIZE; in GetDropoutGenMaskPrim()195 if (cnode_non_monad_size != DROPOUT_GEN_MASK_CNODE_INPUT_SIZE) { in SetGenMaskShape()196 …ON) << "The size of dropout gen mask cnode's inputs must be " << DROPOUT_GEN_MASK_CNODE_INPUT_SIZE; in SetGenMaskShape()
57 constexpr size_t DROPOUT_GEN_MASK_CNODE_INPUT_SIZE = 3; variable