Searched refs:setInfoAndAllocateIfNeeded (Results 1 – 1 of 1) sorted by relevance
/frameworks/ml/nn/common/ |
D | CpuExecutor.cpp | 140 bool setInfoAndAllocateIfNeeded(RunTimeOperandInfo* info, const Shape& shape, int* result) { in setInfoAndAllocateIfNeeded() function 209 return setInfoAndAllocateIfNeeded(getOutputInfo(index), shape, &result); in setOutputShape() 465 if (!setInfoAndAllocateIfNeeded(&to, inShape, &result)) { in convertToNhwc() 503 if (!setInfoAndAllocateIfNeeded(&to, outShape, result)) { in convertFromNhwc() 526 if (!setInfoAndAllocateIfNeeded(&to, outShape, result)) { in convertFromNhwc() 730 !setInfoAndAllocateIfNeeded(&output, outShape, &result)) { in executeOperation() 827 !setInfoAndAllocateIfNeeded(&output_tmp, outShape, &result)) { in executeOperation() 900 !setInfoAndAllocateIfNeeded(&output, outShape, &result)) { in executeOperation() 928 setInfoAndAllocateIfNeeded(&output, outShape, &result) && in executeOperation() 953 !setInfoAndAllocateIfNeeded(&output_tmp, outShape, &result)) { in executeOperation() [all …]
|