Searched refs:DROPOUT_DO_MASK_CNODE_INPUT_SIZE (Results 1 – 2 of 2) sorted by relevance
152 if (cnode->size() != DROPOUT_DO_MASK_CNODE_INPUT_SIZE) { in GetDropoutGenMaskPrim()153 …TION) << "The size of dropout do mask cnode's inputs must be " << DROPOUT_DO_MASK_CNODE_INPUT_SIZE; in GetDropoutGenMaskPrim()183 if (cnode->size() != DROPOUT_DO_MASK_CNODE_INPUT_SIZE) { in SetGenMaskShape()184 …TION) << "The size of dropout do mask cnode's inputs must be " << DROPOUT_DO_MASK_CNODE_INPUT_SIZE; in SetGenMaskShape()229 if (cnode->inputs().size() != DROPOUT_DO_MASK_CNODE_INPUT_SIZE) { in GetDropoutGenMaskReplaceOp()230 …TION) << "The size of dropout do mask cnode's inputs must be " << DROPOUT_DO_MASK_CNODE_INPUT_SIZE; in GetDropoutGenMaskReplaceOp()279 if (node->inputs().size() == DROPOUT_DO_MASK_CNODE_INPUT_SIZE) { in ReplaceOneOp()303 if (cnode->inputs().size() != DROPOUT_DO_MASK_CNODE_INPUT_SIZE) { in ReplaceNodeInputOrAttrs()305 << DROPOUT_DO_MASK_CNODE_INPUT_SIZE; in ReplaceNodeInputOrAttrs()
56 constexpr size_t DROPOUT_DO_MASK_CNODE_INPUT_SIZE = 4; variable