Searched refs:bwOutputShape (Results 1 – 4 of 4) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | BidirectionalSequenceLSTM.cpp | 162 Shape* fwOutputShape, Shape* bwOutputShape) { in Prepare() argument
244 bwOutputShape->type = inputShape.type; in Prepare()
245 bwOutputShape->offset = inputShape.offset; in Prepare()
246 bwOutputShape->scale = inputShape.scale; in Prepare()
247 bwOutputShape->dimensions.resize(3); in Prepare()
248 bwOutputShape->dimensions[0] = params_.time_major ? max_time : n_batch; in Prepare()
249 bwOutputShape->dimensions[1] = params_.time_major ? n_batch : max_time; in Prepare()
250 bwOutputShape->dimensions[2] = n_bw_output; in Prepare()
|
| D | BidirectionalSequenceRNN.cpp | 116 Shape bwOutputShape; in executeTyped() local
118 bwOutputShape = context->getOutputShape(kBwOutputTensor); in executeTyped()
137 bwOutputTransposed.resize(getNumberOfElements(bwOutputShape)); in executeTyped()
164 std::swap(bwOutputShape.dimensions[0], bwOutputShape.dimensions[1]); in executeTyped()
240 transposeFirstTwoDims(bwOutputTransposed.data(), bwOutputShape, in executeTyped()
|
| D | BidirectionalSequenceLSTM.h | 40 Shape* fwOutputShape, Shape* bwOutputShape);
|
| /frameworks/ml/nn/common/ |
| D | CpuExecutor.cpp | 1105 Shape fwOutputShape, bwOutputShape; in executeOperation() local
1108 success = lstm.Prepare(operation, mOperands, &fwOutputShape, &bwOutputShape) && in executeOperation()
1113 success = success && setInfoAndAllocateIfNeeded(&bwOutput, bwOutputShape, &result); in executeOperation()
|