Home
last modified time | relevance | path

Searched refs:shrinkAxisMask (Results 1 – 25 of 40) sorted by relevance

12

/frameworks/ml/nn/runtime/test/specs/V1_3/
Dstrided_slice_quant8_signed.mod.py24 shrinkAxisMask = Int32Scalar("shrinkAxisMask", 0) variable
28 …peration("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
49 shrinkAxisMask = Int32Scalar("shrinkAxisMask", 1) variable
53 …peration("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
74 shrinkAxisMask = Int32Scalar("shrinkAxisMask", 0) variable
78 …peration("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
99 shrinkAxisMask = Int32Scalar("shrinkAxisMask", 0) variable
103 …peration("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
124 shrinkAxisMask = Int32Scalar("shrinkAxisMask", 0) variable
128 …peration("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
[all …]
/frameworks/ml/nn/common/operations/
DStridedSlice.cpp54 int32_t shrinkAxisMask, T* outputData, const Shape& outputShape) { in compute() argument
79 shrinkAxisMask = ReverseMaskBits(shrinkAxisMask, numInputDims); in compute()
82 endMask, shrinkAxisMask, starts, stops, strides, outputData, in compute()
166 const int32_t shrinkAxisMask = context->getInputValue<int32_t>(kShrinkAxisMask); in prepare() local
185 if (!(shrinkAxisMask & (1 << idx))) { in prepare()
/frameworks/ml/nn/runtime/test/specs/V1_1/
Dstrided_slice_qaunt8_11.mod.py8 shrinkAxisMask = Int32Scalar("shrinkAxisMask", 1) variable
12 …peration("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
Dstrided_slice_quant8_9.mod.py8 shrinkAxisMask = Int32Scalar("shrinkAxisMask", 0) variable
12 …peration("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
Dstrided_slice_quant8_1.mod.py8 shrinkAxisMask = Int32Scalar("shrinkAxisMask", 0) variable
12 …peration("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
Dstrided_slice_quant8_8.mod.py8 shrinkAxisMask = Int32Scalar("shrinkAxisMask", 0) variable
12 …peration("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
Dstrided_slice_quant8_6.mod.py8 shrinkAxisMask = Int32Scalar("shrinkAxisMask", 0) variable
12 …peration("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
Dstrided_slice_float_1.mod.py8 shrinkAxisMask = Int32Scalar("shrinkAxisMask", 0) variable
12 …peration("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
Dstrided_slice_float_10.mod.py8 shrinkAxisMask = Int32Scalar("shrinkAxisMask", 0) variable
12 …peration("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
Dstrided_slice_float_3.mod.py8 shrinkAxisMask = Int32Scalar("shrinkAxisMask", 0) variable
12 …peration("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
Dstrided_slice_quant8_2.mod.py8 shrinkAxisMask = Int32Scalar("shrinkAxisMask", 0) variable
12 …peration("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
Dstrided_slice_float_4.mod.py8 shrinkAxisMask = Int32Scalar("shrinkAxisMask", 0) variable
12 …peration("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
Dstrided_slice_float_9.mod.py8 shrinkAxisMask = Int32Scalar("shrinkAxisMask", 0) variable
12 …peration("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
Dstrided_slice_float_5.mod.py8 shrinkAxisMask = Int32Scalar("shrinkAxisMask", 0) variable
12 …peration("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
Dstrided_slice_quant8_3.mod.py8 shrinkAxisMask = Int32Scalar("shrinkAxisMask", 0) variable
12 …peration("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
Dstrided_slice_float_8.mod.py8 shrinkAxisMask = Int32Scalar("shrinkAxisMask", 0) variable
12 …peration("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
Dstrided_slice_quant8_5.mod.py8 shrinkAxisMask = Int32Scalar("shrinkAxisMask", 0) variable
12 …peration("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
Dstrided_slice_quant8_7.mod.py8 shrinkAxisMask = Int32Scalar("shrinkAxisMask", 0) variable
12 …peration("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
Dstrided_slice_float_6.mod.py8 shrinkAxisMask = Int32Scalar("shrinkAxisMask", 0) variable
12 …peration("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
Dstrided_slice.mod.py8 shrinkAxisMask = Int32Scalar("shrinkAxisMask", 0) variable
12 …peration("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
Dstrided_slice_float_2.mod.py8 shrinkAxisMask = Int32Scalar("shrinkAxisMask", 0) variable
12 …peration("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
Dstrided_slice_qaunt8_10.mod.py8 shrinkAxisMask = Int32Scalar("shrinkAxisMask", 0) variable
12 …peration("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
Dstrided_slice_float_11.mod.py8 shrinkAxisMask = Int32Scalar("shrinkAxisMask", 1) variable
12 …peration("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
Dstrided_slice_float_7.mod.py8 shrinkAxisMask = Int32Scalar("shrinkAxisMask", 0) variable
12 …peration("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
/frameworks/ml/nn/runtime/test/specs/V1_2/
Dstrided_slice_float16.mod.py8 shrinkAxisMask = Int32Scalar("shrinkAxisMask", 0) variable
12 …peration("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)

12