Searched defs:fwOutputShape (Results 1 – 3 of 3) sorted by relevance
162 Shape* fwOutputShape, Shape* bwOutputShape) { in Prepare()
114 Shape fwOutputShape = context->getOutputShape(kFwOutputTensor); in executeTyped() local
1105 Shape fwOutputShape, bwOutputShape; in executeOperation() local