Searched refs:bwOutputShape (Results 1 – 4 of 4) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | BidirectionalSequenceRNN.cpp | 167 Shape bwOutputShape; in executeTyped() local
169 bwOutputShape = context->getOutputShape(kBwOutputTensor); in executeTyped()
188 bwOutputTransposed.resize(getNumberOfElements(bwOutputShape)); in executeTyped()
215 std::swap(bwOutputShape.dimensions[0], bwOutputShape.dimensions[1]); in executeTyped()
310 transposeFirstTwoDims(bwOutputTransposed.data(), bwOutputShape, in executeTyped()
|
| D | BidirectionalSequenceLSTM.cpp | 210 Shape* fwOutputShape, Shape* bwOutputShape, in Prepare() argument
410 bwOutputShape->type = inputShape.type; in Prepare()
411 bwOutputShape->offset = inputShape.offset; in Prepare()
412 bwOutputShape->scale = inputShape.scale; in Prepare()
413 bwOutputShape->dimensions.resize(3); in Prepare()
414 bwOutputShape->dimensions[0] = params_.time_major ? max_time : n_batch; in Prepare()
415 bwOutputShape->dimensions[1] = params_.time_major ? n_batch : max_time; in Prepare()
416 bwOutputShape->dimensions[2] = n_bw_output; in Prepare()
|
| D | BidirectionalSequenceLSTM.h | 40 Shape* fwOutputShape, Shape* bwOutputShape, Shape* fwOutputActivationState,
|
| /frameworks/ml/nn/common/ |
| D | CpuExecutor.cpp | 1059 Shape fwOutputShape, bwOutputShape, fwOutputActivationStateShape, in executeOperation() local
1063 success = lstm.Prepare(operation, operands, &fwOutputShape, &bwOutputShape, in executeOperation()
1070 success = success && setInfoAndAllocateIfNeeded(&bwOutput, bwOutputShape, &result); in executeOperation()
|