Home
last modified time | relevance | path

Searched refs:SVDF (Results 1 – 25 of 28) sorted by relevance

12

/packages/modules/NeuralNetworks/common/cpu_operations/
DSVDF.cpp33 SVDF::SVDF(const Operation& operation, RunTimeOperandInfo* operands) { in SVDF() function in android::nn::SVDF
51 bool SVDF::Prepare(const Operation& operation, RunTimeOperandInfo* operands, Shape* stateShape, in Prepare()
74 const RunTimeOperandInfo* input = GetInput(operation, operands, SVDF::kInputTensor); in Prepare()
76 GetInput(operation, operands, SVDF::kWeightsFeatureTensor); in Prepare()
78 GetInput(operation, operands, SVDF::kWeightsTimeTensor); in Prepare()
112 bool SVDF::Eval() { in Eval()
164 void SVDF::EvalFloat32(const float* inputData, const float* inputStateData, const float* biasData, in EvalFloat32()
DSVDFTest.cpp248 ASSERT_EQ(execution.setInput(SVDF::k##X##Tensor, X##_.data(), sizeof(float) * X##_.size()), \ in Invoke()
257 ASSERT_EQ(execution.setOutput(SVDF::k##X##Tensor, X##_.data(), sizeof(float) * X##_.size()), \ in Invoke()
264 ASSERT_EQ(execution.setInput(SVDF::kRankParam, &rank_, sizeof(rank_)), Result::NO_ERROR); in Invoke()
267 ASSERT_EQ(execution.setInput(SVDF::kActivationParam, &activation, sizeof(activation)), in Invoke()
/packages/modules/NeuralNetworks/common/types/operations/include/
DSVDF.h39 class SVDF {
41 SVDF(const Operation& operation, RunTimeOperandInfo* operands);
/packages/modules/NeuralNetworks/runtime/test/generated/spec_V1_1/
Dsvdf2_relaxed.example.cpp104 .type = TestOperationType::SVDF, in get_test_model()
357 .type = TestOperationType::SVDF, in get_test_model_all_inputs_as_internal()
Dsvdf_state_relaxed.example.cpp104 .type = TestOperationType::SVDF, in get_test_model()
357 .type = TestOperationType::SVDF, in get_test_model_all_inputs_as_internal()
Dsvdf_bias_present_relaxed.example.cpp104 .type = TestOperationType::SVDF, in get_test_model()
391 .type = TestOperationType::SVDF, in get_test_model_all_inputs_as_internal()
Dsvdf_relaxed.example.cpp104 .type = TestOperationType::SVDF, in get_test_model()
357 .type = TestOperationType::SVDF, in get_test_model_all_inputs_as_internal()
/packages/modules/NeuralNetworks/runtime/test/generated/spec_V1_0/
Dsvdf_bias_present.example.cpp104 .type = TestOperationType::SVDF, in get_test_model()
391 .type = TestOperationType::SVDF, in get_test_model_all_inputs_as_internal()
Dsvdf2.example.cpp104 .type = TestOperationType::SVDF, in get_test_model()
357 .type = TestOperationType::SVDF, in get_test_model_all_inputs_as_internal()
Dsvdf_state.example.cpp104 .type = TestOperationType::SVDF, in get_test_model()
357 .type = TestOperationType::SVDF, in get_test_model_all_inputs_as_internal()
Dsvdf.example.cpp104 .type = TestOperationType::SVDF, in get_test_model()
357 .type = TestOperationType::SVDF, in get_test_model_all_inputs_as_internal()
/packages/modules/NeuralNetworks/runtime/test/generated/spec_V1_2/
Dsvdf_float16.example.cpp104 .type = TestOperationType::SVDF, in get_test_model()
357 .type = TestOperationType::SVDF, in get_test_model_all_inputs_as_internal()
Dsvdf_state_float16.example.cpp104 .type = TestOperationType::SVDF, in get_test_model()
357 .type = TestOperationType::SVDF, in get_test_model_all_inputs_as_internal()
Dsvdf_bias_present_float16.example.cpp104 .type = TestOperationType::SVDF, in get_test_model()
391 .type = TestOperationType::SVDF, in get_test_model_all_inputs_as_internal()
/packages/modules/NeuralNetworks/common/types/operations/src/
DSVDF.cpp50 NN_DEFINE_VALIDATION_FUNCTION(SVDF, svdf::validate);
/packages/modules/NeuralNetworks/runtime/test/android_fuzzing/
DStaticAssert.cpp83 static_assert(static_cast<TestOperationType>(SVDF) == TestOperationType::SVDF);
DModel.proto68 SVDF = 27; enumerator
/packages/modules/NeuralNetworks/common/types/include/nnapi/
DOperationTypes.h1789 SVDF = 27, enumerator
/packages/modules/NeuralNetworks/runtime/test/android_fuzzing/corpus/
Dseed353137 type: SVDF
/packages/modules/NeuralNetworks/common/
DOperationResolver.cpp41 NN_OPERATION_IS_NOT_IMPLEMENTED(SVDF);
DAndroid.bp164 "cpu_operations/SVDF.cpp",
349 "cpu_operations/SVDF.cpp",
DCpuExecutor.cpp1085 case OperationType::SVDF: { in executeOperation()
1086 RunTimeOperandInfo& stateOut = operands[outs[SVDF::kStateOutTensor]]; in executeOperation()
1087 RunTimeOperandInfo& output = operands[outs[SVDF::kOutputTensor]]; in executeOperation()
1090 SVDF svdf(operation, operands); in executeOperation()
1092 success = SVDF::Prepare(operation, operands, &stateShape, &outputShape) && in executeOperation()
/packages/modules/NeuralNetworks/common/types/include/
DOperationsUtils.h70 CALL(SVDF) \
/packages/modules/NeuralNetworks/runtime/test/
DTestAssertions.cpp70 CHECK_TEST_ENUM(TestOperationType, SVDF);
/packages/modules/NeuralNetworks/tools/test_generator/test_harness/
DAssertions.cpp99 COMPARE_ENUMS(SVDF);

12