Searched refs:setScalarValue (Results 1 – 8 of 8) sorted by relevance
27 op->inputs[2]->setScalarValue<int32_t>(getRandomAxis(rank)); in softmaxConstructor()95 op->inputs[1]->setScalarValue<int32_t>(getRandomAxis(rank)); in l2normConstructor()163 op->inputs[5]->setScalarValue<int32_t>(getRandomAxis(rank)); in localResponseNormConstructor()
326 op->setScalarValue<float>(getUniform<float>(-10.0f, 10.0f)); in __anon5cabe3aa0202()330 op->setScalarValue<_Float16>(getUniform<_Float16>(-10.0f, 10.0f)); in __anon5cabe3aa0202()334 op->setScalarValue<int32_t>(getUniform<int32_t>(0, 255)); in __anon5cabe3aa0202()338 op->setScalarValue<int32_t>(getUniform<int32_t>(-128, 127)); in __anon5cabe3aa0202()429 op->inputs[2]->setScalarValue<int32_t>(axis); in channelShuffleConstructor()523 op->inputs[1]->setScalarValue<int32_t>(axis); in expandDimsConstructor()
118 op->inputs[1]->setScalarValue<int32_t>(axis); in gatherConstructor()302 op->inputs[6]->setScalarValue<int32_t>(convertToBitMask(shrinkMask)); in stridedSliceConstructor()359 op->inputs[4]->setScalarValue<int32_t>(convertToBitMask(beginMask)); in stridedSliceFinalizer()360 op->inputs[5]->setScalarValue<int32_t>(convertToBitMask(endMask)); in stridedSliceFinalizer()
32 op->inputs[numInputs]->setScalarValue<int32_t>(axis); in concatConstructor()124 op->inputs[1]->setScalarValue<int32_t>(axis); in splitConstructor()
107 op->inputs[1]->setScalarValue<int32_t>(axis); in singleAxisReduceOpConstructor()
68 alignCornerFlag->setScalarValue<bool8>(false); in resizeOpConstructor()
71 op->inputs[2]->setScalarValue(0); in broadcastOpConstructor()
99 void setScalarValue(const T& val) { in setScalarValue() function