Searched refs:shrinkMask (Results 1 – 1 of 1) sorted by relevance
290 std::vector<bool> shrinkMask(rank, false); in stridedSliceConstructor() local292 shrinkMask[i] = getBernoulli(0.2f); in stridedSliceConstructor()295 if (!shrinkMask[i]) { in stridedSliceConstructor()302 op->inputs[6]->setScalarValue<int32_t>(convertToBitMask(shrinkMask)); in stridedSliceConstructor()319 int32_t shrinkMask = op->inputs[6]->value<int32_t>(); in stridedSliceFinalizer() local323 bool shrink = shrinkMask & (1 << i); in stridedSliceFinalizer()