Searched refs:setOutputTensor (Results 1 – 1 of 1) sorted by relevance
141 setOutputTensor(&execution, QuantizedLSTMCell::kCellStateOutTensor, &cellStateOut_), in invoke()143 ASSERT_EQ(setOutputTensor(&execution, QuantizedLSTMCell::kOutputTensor, &output_), in invoke()230 Result setOutputTensor(Execution* execution, int tensor, std::vector<T>* data) { in setOutputTensor() function in android::nn::wrapper::QuantizedLSTMOpModel