Searched refs:dimValue (Results 1 – 1 of 1) sorted by relevance
39 uint32_t dimValue = op->inputs[1]->dimensions[0].getValue(); in embeddingLookupFinalizer() local43 op->inputs[0]->value<int32_t>(i) = getUniform<int32_t>(0, dimValue - 1); in embeddingLookupFinalizer()141 uint32_t dimValue = op->inputs[0]->dimensions[axis].getValue(); in gatherFinalizer() local145 op->inputs[2]->value<int32_t>(i) = getUniform<int32_t>(0, dimValue - 1); in gatherFinalizer()