| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | RNN.cpp | 30 RNN::RNN(const Operation& operation, RunTimeOperandInfo* operands) { in RNN() function in android::nn::RNN
45 bool RNN::Prepare(const Operation& operation, RunTimeOperandInfo* operands, Shape* hiddenStateShape, in Prepare()
81 bool RNN::Eval() { in Eval()
116 bool RNN::RNNStep(const T* inputData, const Shape& inputShape, const T* hiddenStateInputData, in RNNStep()
136 bool RNN::RNNStep(const T* inputData, const Shape& inputShape, const T* auxInputData, in RNNStep()
223 template bool RNN::RNNStep<_Float16>(const _Float16* inputData, const Shape& inputShape,
229 template bool RNN::RNNStep<_Float16>(const _Float16* inputData, const Shape& inputShape,
239 template bool RNN::RNNStep<float>(const float* inputData, const Shape& inputShape,
245 template bool RNN::RNNStep<float>(const float* inputData, const Shape& inputShape,
|
| D | RNNTest.cpp | 201 ASSERT_EQ(execution.setInput(RNN::k##X##Tensor, X##_.data(), sizeof(float) * X##_.size()), \ in Invoke()
209 ASSERT_EQ(execution.setOutput(RNN::k##X##Tensor, X##_.data(), sizeof(float) * X##_.size()), \ in Invoke()
216 ASSERT_EQ(execution.setInput(RNN::kActivationParam, &activation_, sizeof(activation_)), in Invoke()
|
| D | BidirectionalSequenceRNN.cpp | 236 RNN::RNNStep<T>(inputBatchPtr, fixedTimeInputShape, auxInputBatchPtr, in executeTyped()
264 RNN::RNNStep<T>(inputBatchPtr, fixedTimeInputShape, auxInputBatchPtr, in executeTyped()
|
| D | UnidirectionalSequenceRNN.cpp | 96 RNN::RNNStep<T>(input, fixedTimeInputShape, hiddenState, bias, weights, weightsShape, in executeTyped()
|
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | RNN.h | 37 class RNN {
39 RNN(const Operation& operation, RunTimeOperandInfo* operands);
|
| /packages/modules/NeuralNetworks/common/types/operations/src/ |
| D | RNN.cpp | 65 NN_DEFINE_VALIDATION_FUNCTION(RNN, rnn::validate);
|
| /packages/modules/NeuralNetworks/runtime/test/generated/spec_V1_1/ |
| D | rnn_relaxed.example.cpp | 94 .type = TestOperationType::RNN, in get_test_model()
371 .type = TestOperationType::RNN, in get_test_model_all_inputs_as_internal()
|
| D | rnn_state_relaxed.example.cpp | 94 .type = TestOperationType::RNN, in get_test_model()
371 .type = TestOperationType::RNN, in get_test_model_all_inputs_as_internal()
|
| /packages/modules/NeuralNetworks/runtime/test/generated/spec_V1_0/ |
| D | rnn.example.cpp | 94 .type = TestOperationType::RNN, in get_test_model()
371 .type = TestOperationType::RNN, in get_test_model_all_inputs_as_internal()
|
| D | rnn_state.example.cpp | 94 .type = TestOperationType::RNN, in get_test_model()
371 .type = TestOperationType::RNN, in get_test_model_all_inputs_as_internal()
|
| /packages/modules/NeuralNetworks/runtime/test/generated/spec_V1_2/ |
| D | rnn_float16.example.cpp | 94 .type = TestOperationType::RNN, in get_test_model()
371 .type = TestOperationType::RNN, in get_test_model_all_inputs_as_internal()
|
| /packages/modules/NeuralNetworks/common/types/include/nnapi/ |
| D | OperationTypes.h | 1619 RNN = 24, enumerator
|
| /packages/modules/NeuralNetworks/runtime/test/android_fuzzing/ |
| D | StaticAssert.cpp | 80 static_assert(static_cast<TestOperationType>(RNN) == TestOperationType::RNN);
|
| D | Model.proto | 65 RNN = 24; enumerator
|
| /packages/modules/NeuralNetworks/common/ |
| D | OperationResolver.cpp | 39 NN_OPERATION_IS_NOT_IMPLEMENTED(RNN);
|
| D | Android.bp | 162 "cpu_operations/RNN.cpp",
349 "cpu_operations/RNN.cpp",
|
| D | CpuExecutor.cpp | 1070 case OperationType::RNN: { in executeOperation()
1075 RunTimeOperandInfo& hiddenStateOut = operands[outs[RNN::kHiddenStateOutTensor]]; in executeOperation()
1076 RunTimeOperandInfo& output = operands[outs[RNN::kOutputTensor]]; in executeOperation()
1079 RNN rnn_cell(operation, operands); in executeOperation()
1081 success = RNN::Prepare(operation, operands, &hiddenStateShape, &outputShape) && in executeOperation()
|
| /packages/modules/NeuralNetworks/runtime/operation_converters/ |
| D | OperationConverterResolver.cpp | 53 NN_OPERATION_CONVERTER_NOT_IMPLEMENTED(RNN);
|
| /packages/modules/NeuralNetworks/common/types/include/ |
| D | OperationsUtils.h | 67 CALL(RNN) \
|
| /packages/modules/NeuralNetworks/runtime/test/ |
| D | TestAssertions.cpp | 67 CHECK_TEST_ENUM(TestOperationType, RNN);
|
| /packages/modules/NeuralNetworks/tools/test_generator/test_harness/ |
| D | Assertions.cpp | 96 COMPARE_ENUMS(RNN);
|
| /packages/modules/NeuralNetworks/tools/test_generator/test_harness/include/ |
| D | TestHarness.h | 119 RNN = 24, enumerator
|
| /packages/modules/NeuralNetworks/runtime/ |
| D | NeuralNetworksV2.cpp | 342 static_assert(static_cast<int32_t>(OperationType::RNN) == ANEURALNETWORKS_RNN,
|
| D | NeuralNetworks.cpp | 340 static_assert(static_cast<int32_t>(OperationType::RNN) == ANEURALNETWORKS_RNN,
|
| /packages/modules/NeuralNetworks/tools/api/ |
| D | types.spec | 2276 %{DeclareOperation RNN 24},
3276 * one input into the two RNN cells in the following way:
3286 * An op with cross-linking takes two inputs and feeds them into the RNN
3308 * In this case, the cell feeds inputs into the RNN in the following way:
6016 * A recurrent neural network layer that applies a basic RNN cell to a
|