Home
last modified time | relevance | path

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

1234

/frameworks/ml/nn/common/operations/
DStridedSlice.cpp34 int32_t shrinkAxisMask, uint8_t* outputData, const Shape& outputShape) { in stridedSliceGeneric() argument
59 shrinkAxisMask = ReverseMaskBits(shrinkAxisMask, numInputDims); in stridedSliceGeneric()
65 beginMask, endMask, shrinkAxisMask, starts, stops, strides, in stridedSliceGeneric()
71 beginMask, endMask, shrinkAxisMask, starts, stops, strides, in stridedSliceGeneric()
77 beginMask, endMask, shrinkAxisMask, starts, stops, strides, in stridedSliceGeneric()
/frameworks/ml/nn/runtime/test/generated/models/
Dstrided_slice_float_3.model.cpp15 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel() local
29 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel()
30 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel()
55 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
69 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel_dynamic_output_shape()
70 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel_dynamic_output_shape()
Dstrided_slice_float_6_relaxed.model.cpp15 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel() local
29 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel()
30 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel()
57 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
71 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel_dynamic_output_shape()
72 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel_dynamic_output_shape()
Dstrided_slice_float_7_relaxed.model.cpp14 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel() local
28 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel()
29 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel()
56 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
70 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel_dynamic_output_shape()
71 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel_dynamic_output_shape()
Dstrided_slice_relaxed.model.cpp15 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel() local
29 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel()
30 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel()
57 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
71 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel_dynamic_output_shape()
72 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel_dynamic_output_shape()
Dstrided_slice_float_9_relaxed.model.cpp15 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel() local
29 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel()
30 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel()
57 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
71 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel_dynamic_output_shape()
72 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel_dynamic_output_shape()
Dstrided_slice_quant8_7.model.cpp14 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel() local
28 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel()
29 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel()
54 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
68 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel_dynamic_output_shape()
69 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel_dynamic_output_shape()
Dstrided_slice_qaunt8_10.model.cpp15 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel() local
29 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel()
30 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel()
55 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
69 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel_dynamic_output_shape()
70 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel_dynamic_output_shape()
Dstrided_slice_float_1_relaxed.model.cpp15 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel() local
29 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel()
30 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel()
57 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
71 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel_dynamic_output_shape()
72 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel_dynamic_output_shape()
Dstrided_slice_float16.model.cpp15 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel() local
29 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel()
30 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel()
55 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
69 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel_dynamic_output_shape()
70 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel_dynamic_output_shape()
Dstrided_slice_float_5.model.cpp15 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel() local
29 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel()
30 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel()
55 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
69 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel_dynamic_output_shape()
70 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel_dynamic_output_shape()
Dstrided_slice_float_9.model.cpp15 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel() local
29 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel()
30 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel()
55 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
69 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel_dynamic_output_shape()
70 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel_dynamic_output_shape()
Dstrided_slice_float_2_relaxed.model.cpp15 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel() local
29 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel()
30 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel()
57 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
71 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel_dynamic_output_shape()
72 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel_dynamic_output_shape()
Dstrided_slice_quant8_9.model.cpp15 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel() local
29 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel()
30 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel()
55 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
69 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel_dynamic_output_shape()
70 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel_dynamic_output_shape()
Dstrided_slice_float_4.model.cpp15 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel() local
29 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel()
30 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel()
55 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
69 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel_dynamic_output_shape()
70 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel_dynamic_output_shape()
Dstrided_slice_quant8_4.model.cpp15 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel() local
29 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel()
30 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel()
55 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
69 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel_dynamic_output_shape()
70 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel_dynamic_output_shape()
Dstrided_slice.model.cpp15 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel() local
29 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel()
30 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel()
55 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
69 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel_dynamic_output_shape()
70 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel_dynamic_output_shape()
Dstrided_slice_float_1.model.cpp15 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel() local
29 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel()
30 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel()
55 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
69 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel_dynamic_output_shape()
70 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel_dynamic_output_shape()
Dstrided_slice_float_3_relaxed.model.cpp15 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel() local
29 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel()
30 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel()
57 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
71 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel_dynamic_output_shape()
72 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel_dynamic_output_shape()
Dstrided_slice_float_11.model.cpp15 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel() local
29 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel()
30 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel()
55 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
69 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel_dynamic_output_shape()
70 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel_dynamic_output_shape()
Dstrided_slice_float_5_relaxed.model.cpp15 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel() local
29 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel()
30 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel()
57 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
71 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel_dynamic_output_shape()
72 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel_dynamic_output_shape()
Dstrided_slice_float_2.model.cpp15 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel() local
29 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel()
30 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel()
55 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
69 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel_dynamic_output_shape()
70 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel_dynamic_output_shape()
Dstrided_slice_float_10_relaxed.model.cpp15 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel() local
29 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel()
30 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel()
57 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
71 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel_dynamic_output_shape()
72 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel_dynamic_output_shape()
Dstrided_slice_float_7.model.cpp14 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel() local
28 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel()
29 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel()
54 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
68 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel_dynamic_output_shape()
69 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel_dynamic_output_shape()
Dstrided_slice_quant8_5.model.cpp15 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel() local
29 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel()
30 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel()
55 auto shrinkAxisMask = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
69 model->setOperandValue(shrinkAxisMask, shrinkAxisMask_init, sizeof(int32_t) * 1); in CreateModel_dynamic_output_shape()
70 …WORKS_STRIDED_SLICE, {input, begins, ends, strides, beginMask, endMask, shrinkAxisMask}, {output}); in CreateModel_dynamic_output_shape()

1234