Searched refs:setScalarValue (Results 1 – 6 of 6) sorted by relevance
27 op->inputs[2]->setScalarValue<int32_t>(getUniform<int32_t>(-rank, rank - 1)); in softmaxConstructor()72 op->inputs[1]->setScalarValue<int32_t>(getUniform<int32_t>(-rank, rank - 1)); in l2normConstructor()118 op->inputs[5]->setScalarValue<int32_t>(getUniform<int32_t>(-rank, rank - 1)); in localResponseNormConstructor()
32 op->inputs[numInputs]->setScalarValue<int32_t>(axis); in concatConstructor()95 op->inputs[1]->setScalarValue<int32_t>(axis); in splitConstructor()
105 op->inputs[1]->setScalarValue<int32_t>(axis); in gatherConstructor()264 op->inputs[6]->setScalarValue<int32_t>(convertToBitMask(shrinkMask)); in stridedSliceConstructor()311 op->inputs[4]->setScalarValue<int32_t>(convertToBitMask(beginMask)); in stridedSliceFinalizer()312 op->inputs[5]->setScalarValue<int32_t>(convertToBitMask(endMask)); in stridedSliceFinalizer()
314 op->setScalarValue<float>(getUniform<float>(-10.0f, 10.0f)); in __anon8e0f86a60202()318 op->setScalarValue<_Float16>(getUniform<_Float16>(-10.0f, 10.0f)); in __anon8e0f86a60202()322 op->setScalarValue<int32_t>(getUniform<int32_t>(0, 255)); in __anon8e0f86a60202()397 op->inputs[2]->setScalarValue<int32_t>(axis); in channelShuffleConstructor()460 op->inputs[1]->setScalarValue<int32_t>(axis); in expandDimsConstructor()
102 op->inputs[1]->setScalarValue<int32_t>(axis); in singleAxisReduceOpConstructor()
88 void setScalarValue(const T& val) { in setScalarValue() function