/frameworks/ml/nn/common/include/ |
H A D | CpuOperationUtils.h | 43 dims.strides[0] = 1; 45 dims.strides[i] = dims.strides[i-1] * dims.sizes[i-1];
|
/frameworks/ml/nn/runtime/test/specs/V1_1/ |
H A D | strided_slice.mod.py | 5 strides = Parameter("strides", "TENSOR_INT32", "{2}", [2, 2]) variable 12 model = model.Operation("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
|
H A D | strided_slice_float_1.mod.py | 5 strides = Parameter("strides", "TENSOR_INT32", "{1}", [1]) variable 12 model = model.Operation("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
|
H A D | strided_slice_float_10.mod.py | 5 strides = Parameter("strides", "TENSOR_INT32", "{2}", [1, 1]) variable 12 model = model.Operation("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
|
H A D | strided_slice_float_10_relaxed.mod.py | 21 strides = Parameter("strides", "TENSOR_INT32", "{2}", [1, 1]) variable 28 model = model.Operation("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
|
H A D | strided_slice_float_11.mod.py | 5 strides = Parameter("strides", "TENSOR_INT32", "{2}", [1, 1]) variable 12 model = model.Operation("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
|
H A D | strided_slice_float_11_relaxed.mod.py | 21 strides = Parameter("strides", "TENSOR_INT32", "{2}", [1, 1]) variable 28 model = model.Operation("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
|
H A D | strided_slice_float_1_relaxed.mod.py | 21 strides = Parameter("strides", "TENSOR_INT32", "{1}", [1]) variable 28 model = model.Operation("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
|
H A D | strided_slice_float_2.mod.py | 5 strides = Parameter("strides", "TENSOR_INT32", "{1}", [1]) variable 12 model = model.Operation("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
|
H A D | strided_slice_float_2_relaxed.mod.py | 21 strides = Parameter("strides", "TENSOR_INT32", "{1}", [1]) variable 28 model = model.Operation("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
|
H A D | strided_slice_float_3.mod.py | 5 strides = Parameter("strides", "TENSOR_INT32", "{1}", [1]) variable 12 model = model.Operation("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
|
H A D | strided_slice_float_3_relaxed.mod.py | 21 strides = Parameter("strides", "TENSOR_INT32", "{1}", [1]) variable 28 model = model.Operation("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
|
H A D | strided_slice_float_4.mod.py | 5 strides = Parameter("strides", "TENSOR_INT32", "{1}", [1]) variable 12 model = model.Operation("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
|
H A D | strided_slice_float_4_relaxed.mod.py | 21 strides = Parameter("strides", "TENSOR_INT32", "{1}", [1]) variable 28 model = model.Operation("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
|
H A D | strided_slice_float_5.mod.py | 5 strides = Parameter("strides", "TENSOR_INT32", "{1}", [1]) variable 12 model = model.Operation("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
|
H A D | strided_slice_float_5_relaxed.mod.py | 21 strides = Parameter("strides", "TENSOR_INT32", "{1}", [1]) variable 28 model = model.Operation("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
|
H A D | strided_slice_float_6.mod.py | 5 strides = Parameter("strides", "TENSOR_INT32", "{1}", [1]) variable 12 model = model.Operation("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
|
H A D | strided_slice_float_6_relaxed.mod.py | 21 strides = Parameter("strides", "TENSOR_INT32", "{1}", [1]) variable 28 model = model.Operation("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
|
H A D | strided_slice_float_7.mod.py | 5 strides = Parameter("strides", "TENSOR_INT32", "{1}", [-1]) variable 12 model = model.Operation("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
|
H A D | strided_slice_float_7_relaxed.mod.py | 21 strides = Parameter("strides", "TENSOR_INT32", "{1}", [-1]) variable 28 model = model.Operation("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
|
H A D | strided_slice_float_8.mod.py | 5 strides = Parameter("strides", "TENSOR_INT32", "{2}", [2, -1]) variable 12 model = model.Operation("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
|
H A D | strided_slice_float_8_relaxed.mod.py | 21 strides = Parameter("strides", "TENSOR_INT32", "{2}", [2, -1]) variable 28 model = model.Operation("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
|
H A D | strided_slice_float_9.mod.py | 5 strides = Parameter("strides", "TENSOR_INT32", "{2}", [1, 1]) variable 12 model = model.Operation("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
|
H A D | strided_slice_float_9_relaxed.mod.py | 21 strides = Parameter("strides", "TENSOR_INT32", "{2}", [1, 1]) variable 28 model = model.Operation("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
|
H A D | strided_slice_qaunt8_10.mod.py | 5 strides = Parameter("strides", "TENSOR_INT32", "{2}", [1, 1]) variable 12 model = model.Operation("STRIDED_SLICE", i1, begins, ends, strides, beginMask, endMask, shrinkAxisMask).To(output)
|