Searched refs:setInfoAndAllocateIfNeeded (Results 1 – 1 of 1) sorted by relevance
| /packages/modules/NeuralNetworks/common/ |
| D | CpuExecutor.cpp | 168 bool setInfoAndAllocateIfNeeded(RunTimeOperandInfo* info, const Shape& shape, int* result) { in setInfoAndAllocateIfNeeded() function
245 return setInfoAndAllocateIfNeeded(getOutputInfo(index), shape, &result); in setOutputShape()
449 if (!setInfoAndAllocateIfNeeded(&to, inShape, &result)) { in convertToNhwc()
490 if (!setInfoAndAllocateIfNeeded(&to, outShape, result)) { in convertFromNhwc()
517 if (!setInfoAndAllocateIfNeeded(&to, outShape, result)) { in convertFromNhwc()
808 setInfoAndAllocateIfNeeded(&output, outShape, &result) && in executeOperation()
833 !setInfoAndAllocateIfNeeded(&output_tmp, outShape, &result)) { in executeOperation()
898 !setInfoAndAllocateIfNeeded(&output_tmp, outShape, &result)) { in executeOperation()
952 setInfoAndAllocateIfNeeded(&output, outputShape, &result) && lookup.Eval(); in executeOperation()
970 setInfoAndAllocateIfNeeded(&output, outputShape, &result) && in executeOperation()
[all …]
|