Searched refs:outDim (Results 1 – 3 of 3) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | StridedSlice.cpp | 142 int32_t outDim = ceil((end - begin) / static_cast<float>(stride)); in prepare() local
143 outDim = outDim < 0 ? 0 : static_cast<uint32_t>(outDim); in prepare()
145 outDims.push_back(outDim); in prepare()
149 NN_RET_CHECK_EQ(outDim, 1) << "index = " << idx; in prepare()
|
| /packages/modules/NeuralNetworks/runtime/test/fuzzing/operation_signatures/ |
| D | ConcatSplit.cpp | 135 RandomVariable outDim; in splitConstructor() local
137 outDim = op->inputs[0]->dimensions[i].exactDiv(numSplits); in splitConstructor()
139 outDim = op->inputs[0]->dimensions[i]; in splitConstructor()
141 for (uint32_t j = 0; j < numSplits; j++) op->outputs[j]->dimensions[i] = outDim; in splitConstructor()
|
| /packages/modules/NeuralNetworks/runtime/test/ |
| D | TestValidateOperations.cpp | 3230 uint32_t outDim[] = {4, 2, 2, 1}; in roiAlignOpTest() local
3239 {getOpType(inputOperandCode, 4, outDim)}); in roiAlignOpTest()
3265 uint32_t outDim[] = {4, 2, 2, 1}; in roiPoolingOpTest() local
3273 {getOpType(inputOperandCode, 4, outDim)}); in roiPoolingOpTest()
3353 uint32_t outDim[] = {1, 2, 2, 2}; in groupedConvOpTest() local
3379 ANeuralNetworksOperandType output = getOpType(inputOperandCode, 4, outDim); in groupedConvOpTest()
3428 uint32_t outDim[] = {1, 5, 5, 2}, outShapeDim[] = {4}; in transposeConvOpTest() local
3452 ANeuralNetworksOperandType output = getOpType(inputOperandCode, 4, outDim); in transposeConvOpTest()
3660 uint32_t outDim[] = {5, 8}; in axisAlignedBBoxTransformOpTest() local
3666 {getOpType(roiOperandCode, 2, outDim)}); in axisAlignedBBoxTransformOpTest()
[all …]
|