| /frameworks/ml/nn/runtime/test/generated/models/ |
| D | squeeze_quant8_1.model.cpp | 9 auto squeezeDims = model->addOperand(&type1); in CreateModel() local
13 model->setOperandValue(squeezeDims, squeezeDims_init, sizeof(int32_t) * 1); in CreateModel()
14 model->addOperation(ANEURALNETWORKS_SQUEEZE, {input, squeezeDims}, {output}); in CreateModel()
33 auto squeezeDims = model->addOperand(&type1); in CreateModel_dynamic_output_shape() local
37 model->setOperandValue(squeezeDims, squeezeDims_init, sizeof(int32_t) * 1); in CreateModel_dynamic_output_shape()
38 model->addOperation(ANEURALNETWORKS_SQUEEZE, {input, squeezeDims}, {output}); in CreateModel_dynamic_output_shape()
|
| D | squeeze_float_1_relaxed.model.cpp | 9 auto squeezeDims = model->addOperand(&type1); in CreateModel() local
13 model->setOperandValue(squeezeDims, squeezeDims_init, sizeof(int32_t) * 1); in CreateModel()
14 model->addOperation(ANEURALNETWORKS_SQUEEZE, {input, squeezeDims}, {output}); in CreateModel()
35 auto squeezeDims = model->addOperand(&type1); in CreateModel_dynamic_output_shape() local
39 model->setOperandValue(squeezeDims, squeezeDims_init, sizeof(int32_t) * 1); in CreateModel_dynamic_output_shape()
40 model->addOperation(ANEURALNETWORKS_SQUEEZE, {input, squeezeDims}, {output}); in CreateModel_dynamic_output_shape()
|
| D | squeeze.model.cpp | 9 auto squeezeDims = model->addOperand(&type1); in CreateModel() local
13 model->setOperandValue(squeezeDims, squeezeDims_init, sizeof(int32_t) * 2); in CreateModel()
14 model->addOperation(ANEURALNETWORKS_SQUEEZE, {input, squeezeDims}, {output}); in CreateModel()
33 auto squeezeDims = model->addOperand(&type1); in CreateModel_dynamic_output_shape() local
37 model->setOperandValue(squeezeDims, squeezeDims_init, sizeof(int32_t) * 2); in CreateModel_dynamic_output_shape()
38 model->addOperation(ANEURALNETWORKS_SQUEEZE, {input, squeezeDims}, {output}); in CreateModel_dynamic_output_shape()
|
| D | squeeze_float16.model.cpp | 9 auto squeezeDims = model->addOperand(&type1); in CreateModel() local
13 model->setOperandValue(squeezeDims, squeezeDims_init, sizeof(int32_t) * 2); in CreateModel()
14 model->addOperation(ANEURALNETWORKS_SQUEEZE, {input, squeezeDims}, {output}); in CreateModel()
33 auto squeezeDims = model->addOperand(&type1); in CreateModel_dynamic_output_shape() local
37 model->setOperandValue(squeezeDims, squeezeDims_init, sizeof(int32_t) * 2); in CreateModel_dynamic_output_shape()
38 model->addOperation(ANEURALNETWORKS_SQUEEZE, {input, squeezeDims}, {output}); in CreateModel_dynamic_output_shape()
|
| D | squeeze_relaxed.model.cpp | 9 auto squeezeDims = model->addOperand(&type1); in CreateModel() local
13 model->setOperandValue(squeezeDims, squeezeDims_init, sizeof(int32_t) * 2); in CreateModel()
14 model->addOperation(ANEURALNETWORKS_SQUEEZE, {input, squeezeDims}, {output}); in CreateModel()
35 auto squeezeDims = model->addOperand(&type1); in CreateModel_dynamic_output_shape() local
39 model->setOperandValue(squeezeDims, squeezeDims_init, sizeof(int32_t) * 2); in CreateModel_dynamic_output_shape()
40 model->addOperation(ANEURALNETWORKS_SQUEEZE, {input, squeezeDims}, {output}); in CreateModel_dynamic_output_shape()
|
| D | squeeze_float_1.model.cpp | 9 auto squeezeDims = model->addOperand(&type1); in CreateModel() local
13 model->setOperandValue(squeezeDims, squeezeDims_init, sizeof(int32_t) * 1); in CreateModel()
14 model->addOperation(ANEURALNETWORKS_SQUEEZE, {input, squeezeDims}, {output}); in CreateModel()
33 auto squeezeDims = model->addOperand(&type1); in CreateModel_dynamic_output_shape() local
37 model->setOperandValue(squeezeDims, squeezeDims_init, sizeof(int32_t) * 1); in CreateModel_dynamic_output_shape()
38 model->addOperation(ANEURALNETWORKS_SQUEEZE, {input, squeezeDims}, {output}); in CreateModel_dynamic_output_shape()
|
| /frameworks/ml/nn/runtime/test/specs/V1_1/ |
| D | squeeze.mod.py | 3 squeezeDims = Parameter("squeezeDims", "TENSOR_INT32", "{2}", [1, 2]) variable
6 model = model.Operation("SQUEEZE", i1, squeezeDims).To(output)
|
| D | squeeze_float_1.mod.py | 3 squeezeDims = Parameter("squeezeDims", "TENSOR_INT32", "{1}", [2]) variable
6 model = model.Operation("SQUEEZE", i1, squeezeDims).To(output)
|
| D | squeeze_quant8_1.mod.py | 3 squeezeDims = Parameter("squeezeDims", "TENSOR_INT32", "{1}", [2]) variable
6 model = model.Operation("SQUEEZE", i1, squeezeDims).To(output)
|
| D | squeeze_float_1_relaxed.mod.py | 19 squeezeDims = Parameter("squeezeDims", "TENSOR_INT32", "{1}", [2]) variable
22 model = model.Operation("SQUEEZE", i1, squeezeDims).To(output)
|
| D | squeeze_relaxed.mod.py | 19 squeezeDims = Parameter("squeezeDims", "TENSOR_INT32", "{2}", [1, 2]) variable
22 model = model.Operation("SQUEEZE", i1, squeezeDims).To(output)
|
| /frameworks/ml/nn/runtime/test/specs/V1_2/ |
| D | squeeze_float16.mod.py | 3 squeezeDims = Parameter("squeezeDims", "TENSOR_INT32", "{2}", [1, 2]) variable
6 model = model.Operation("SQUEEZE", i1, squeezeDims).To(output)
|
| /frameworks/ml/nn/common/ |
| D | OperationsUtils.cpp | 646 const int32_t* squeezeDims, in squeezePrepare() argument
669 int32_t current = squeezeDims[idx] < 0 ? squeezeDims[idx] + numInputDims in squeezePrepare()
670 : squeezeDims[idx]; in squeezePrepare()
|
| D | CpuExecutor.cpp | 1358 const RunTimeOperandInfo& squeezeDims = mOperands[ins[1]]; in executeOperation() local
1364 reinterpret_cast<const int32_t*>(squeezeDims.buffer), in executeOperation()
1365 squeezeDims.shape(), &outShape) && in executeOperation()
|
| /frameworks/ml/nn/common/include/ |
| D | OperationsUtils.h | 347 const int32_t* squeezeDims,
|