Searched refs:fwOutputShape (Results 1 – 4 of 4) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | BidirectionalSequenceLSTM.cpp | 162 Shape* fwOutputShape, Shape* bwOutputShape) { in Prepare() argument
221 fwOutputShape->type = inputShape.type; in Prepare()
222 fwOutputShape->offset = inputShape.offset; in Prepare()
223 fwOutputShape->scale = inputShape.scale; in Prepare()
224 fwOutputShape->dimensions.resize(3); in Prepare()
225 fwOutputShape->dimensions[0] = params_.time_major ? max_time : n_batch; in Prepare()
226 fwOutputShape->dimensions[1] = params_.time_major ? n_batch : max_time; in Prepare()
227 fwOutputShape->dimensions[2] = params_.merge_outputs ? n_fw_output + n_bw_output : n_fw_output; in Prepare()
|
| D | BidirectionalSequenceRNN.cpp | 114 Shape fwOutputShape = context->getOutputShape(kFwOutputTensor); in executeTyped() local
135 fwOutputTransposed.resize(getNumberOfElements(fwOutputShape)); in executeTyped()
162 std::swap(fwOutputShape.dimensions[0], fwOutputShape.dimensions[1]); in executeTyped()
237 transposeFirstTwoDims(fwOutputTransposed.data(), fwOutputShape, 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()
1109 setInfoAndAllocateIfNeeded(&fwOutput, fwOutputShape, &result); in executeOperation()
|