aboutsummaryrefslogtreecommitdiff
path: root/src/runtime/cpu/operators/CpuWinogradConv2d.h
diff options
context:
space:
mode:
Diffstat (limited to 'src/runtime/cpu/operators/CpuWinogradConv2d.h')
-rw-r--r--src/runtime/cpu/operators/CpuWinogradConv2d.h27
1 files changed, 13 insertions, 14 deletions
diff --git a/src/runtime/cpu/operators/CpuWinogradConv2d.h b/src/runtime/cpu/operators/CpuWinogradConv2d.h
index ae705ac86b..b5b9c3f2e3 100644
--- a/src/runtime/cpu/operators/CpuWinogradConv2d.h
+++ b/src/runtime/cpu/operators/CpuWinogradConv2d.h
@@ -93,20 +93,19 @@ public:
private:
enum AuxTensorIdx
{
- GemmWorkspace = 0,
- Pretranspose,
- InterleavedLHS,
- TransposedRHS,
- TempResult,
- PermutedInput,
- InputTransformed,
- InputWorkspace,
- PermutedOutput,
- PermutedWeights,
- WeightsTransformed,
- OutputTransformed,
- OutputWorkspace,
- Count
+ GemmWorkspace = 0,
+ Pretranspose = 1,
+ InterleavedLHS = 2,
+ TransposedRHS = 3,
+ TempResult = 4,
+ TransformedInput = 5,
+ TransformedOutput = 6,
+ WorkspaceIO = 7,
+ TransformedWeights = 8,
+ PermutedWeights = 9,
+ PermutedInput = TransformedOutput,
+ PermutedOutput = TransformedInput,
+ Count = 10
};
std::unique_ptr<CpuGemm> _gemm_function;