| /frameworks/ml/nn/runtime/test/specs/V1_1/ |
| H A D | squeeze.mod.py | 3 squeezeDims = Parameter("squeezeDims", "TENSOR_INT32", "{2}", [1, 2]) variable
6 model = model.Operation("SQUEEZE", i1, squeezeDims).To(output)
|
| H A D | squeeze_float_1.mod.py | 3 squeezeDims = Parameter("squeezeDims", "TENSOR_INT32", "{1}", [2]) variable
6 model = model.Operation("SQUEEZE", i1, squeezeDims).To(output)
|
| H A 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)
|
| H A D | squeeze_quant8_1.mod.py | 3 squeezeDims = Parameter("squeezeDims", "TENSOR_INT32", "{1}", [2]) variable
6 model = model.Operation("SQUEEZE", i1, squeezeDims).To(output)
|
| H A 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/generated/models/ |
| H A D | squeeze.model.cpp | 8 auto squeezeDims = model->addOperand(&type1); local
12 model->setOperandValue(squeezeDims, squeezeDims_init, sizeof(int32_t) * 2);
13 model->addOperation(ANEURALNETWORKS_SQUEEZE, {input, squeezeDims}, {output});
|
| H A D | squeeze_float_1.model.cpp | 8 auto squeezeDims = model->addOperand(&type1); local
12 model->setOperandValue(squeezeDims, squeezeDims_init, sizeof(int32_t) * 1);
13 model->addOperation(ANEURALNETWORKS_SQUEEZE, {input, squeezeDims}, {output});
|
| H A D | squeeze_float_1_relaxed.model.cpp | 8 auto squeezeDims = model->addOperand(&type1); local
12 model->setOperandValue(squeezeDims, squeezeDims_init, sizeof(int32_t) * 1);
13 model->addOperation(ANEURALNETWORKS_SQUEEZE, {input, squeezeDims}, {output});
|
| H A D | squeeze_quant8_1.model.cpp | 8 auto squeezeDims = model->addOperand(&type1); local
12 model->setOperandValue(squeezeDims, squeezeDims_init, sizeof(int32_t) * 1);
13 model->addOperation(ANEURALNETWORKS_SQUEEZE, {input, squeezeDims}, {output});
|
| H A D | squeeze_relaxed.model.cpp | 8 auto squeezeDims = model->addOperand(&type1); local
12 model->setOperandValue(squeezeDims, squeezeDims_init, sizeof(int32_t) * 2);
13 model->addOperation(ANEURALNETWORKS_SQUEEZE, {input, squeezeDims}, {output});
|
| /frameworks/ml/nn/common/ |
| H A D | OperationsUtils.cpp | 673 const int32_t* squeezeDims,
678 // squeezeDims need to be provided as a 1-D int32 tensor.
696 int32_t current = squeezeDims[idx] < 0 ? squeezeDims[idx] + numInputDims
697 : squeezeDims[idx];
672 squeezePrepare(const Shape& input, const int32_t* squeezeDims, const Shape& squeezeDimsShape, Shape* output) argument
|
| H A D | CpuExecutor.cpp | 1382 const RunTimeOperandInfo& squeezeDims = mOperands[ins[1]]; local
1388 reinterpret_cast<const int32_t*>(squeezeDims.buffer),
1389 squeezeDims.shape(),
|
| /frameworks/ml/nn/common/include/ |
| H A D | OperationsUtils.h | 261 const int32_t* squeezeDims,
|