| /frameworks/ml/nn/common/operations/ |
| D | SVDF.cpp | 28 SVDF::SVDF(const Operation& operation, in SVDF() function in android::nn::SVDF
45 bool SVDF::Prepare(const Operation &operation, in Prepare()
57 GetInput(operation, operands, SVDF::kInputTensor); in Prepare()
59 GetInput(operation, operands, SVDF::kWeightsFeatureTensor); in Prepare()
61 GetInput(operation, operands, SVDF::kWeightsTimeTensor); in Prepare()
96 bool SVDF::Eval() { in Eval()
148 void SVDF::EvalFloat32(const float* inputData, const float* inputStateData, const float* biasData, in EvalFloat32()
|
| D | SVDF.h | 37 class SVDF {
39 SVDF(const Operation& operation, std::vector<RunTimeOperandInfo>& operands);
|
| D | SVDFTest.cpp | 264 ASSERT_EQ(execution.setInput(SVDF::k##X##Tensor, X##_.data(), \ in Invoke()
274 ASSERT_EQ(execution.setOutput(SVDF::k##X##Tensor, X##_.data(), \ in Invoke()
282 ASSERT_EQ(execution.setInput(SVDF::kRankParam, &rank_, sizeof(rank_)), in Invoke()
286 ASSERT_EQ(execution.setInput(SVDF::kActivationParam, &activation, in Invoke()
|
| /frameworks/ml/nn/runtime/test/generated/vts_models/ |
| D | svdf_state_float16.model.cpp | 91 .type = OperationType::SVDF, in createTestModel()
207 .type = OperationType::SVDF, in createTestModel_dynamic_output_shape()
|
| D | svdf_state.model.cpp | 91 .type = OperationType::SVDF, in createTestModel()
207 .type = OperationType::SVDF, in createTestModel_dynamic_output_shape()
|
| D | svdf_float16.model.cpp | 91 .type = OperationType::SVDF, in createTestModel()
207 .type = OperationType::SVDF, in createTestModel_dynamic_output_shape()
|
| D | svdf.model.cpp | 91 .type = OperationType::SVDF, in createTestModel()
207 .type = OperationType::SVDF, in createTestModel_dynamic_output_shape()
|
| D | svdf2.model.cpp | 91 .type = OperationType::SVDF, in createTestModel()
207 .type = OperationType::SVDF, in createTestModel_dynamic_output_shape()
|
| D | svdf_bias_present.model.cpp | 91 .type = OperationType::SVDF, in createTestModel()
207 .type = OperationType::SVDF, in createTestModel_dynamic_output_shape()
|
| D | svdf_state_relaxed.model.cpp | 91 .type = OperationType::SVDF, in createTestModel()
208 .type = OperationType::SVDF, in createTestModel_dynamic_output_shape()
|
| D | svdf2_relaxed.model.cpp | 91 .type = OperationType::SVDF, in createTestModel()
208 .type = OperationType::SVDF, in createTestModel_dynamic_output_shape()
|
| D | svdf_bias_present_float16.model.cpp | 91 .type = OperationType::SVDF, in createTestModel()
207 .type = OperationType::SVDF, in createTestModel_dynamic_output_shape()
|
| D | svdf_relaxed.model.cpp | 91 .type = OperationType::SVDF, in createTestModel()
208 .type = OperationType::SVDF, in createTestModel_dynamic_output_shape()
|
| D | svdf_bias_present_relaxed.model.cpp | 91 .type = OperationType::SVDF, in createTestModel()
208 .type = OperationType::SVDF, in createTestModel_dynamic_output_shape()
|
| /frameworks/ml/nn/common/ |
| D | Android.bp | 150 "operations/SVDF.cpp",
|
| D | CpuExecutor.cpp | 1157 case OperationType::SVDF: { in executeOperation()
1158 RunTimeOperandInfo& stateOut = mOperands[outs[SVDF::kStateOutTensor]]; in executeOperation()
1159 RunTimeOperandInfo& output = mOperands[outs[SVDF::kOutputTensor]]; in executeOperation()
1162 SVDF svdf(operation, mOperands); in executeOperation()
1164 success = SVDF::Prepare(operation, mOperands, &stateShape, &outputShape) && in executeOperation()
|
| /frameworks/ml/nn/runtime/ |
| D | NeuralNetworks.cpp | 309 static_assert(static_cast<int32_t>(OperationType::SVDF) == ANEURALNETWORKS_SVDF,
|