| /frameworks/ml/nn/common/operations/ |
| SVDFTest.cpp | 17 #include "SVDF.h"
183 // Derived class of SingleOpModel, which is used to test SVDF TFLite op.
264 ASSERT_EQ(execution.setInput(SVDF::k##X##Tensor, X##_.data(), \
274 ASSERT_EQ(execution.setOutput(SVDF::k##X##Tensor, X##_.data(), \
282 ASSERT_EQ(execution.setInput(SVDF::kRankParam, &rank_, sizeof(rank_)),
286 ASSERT_EQ(execution.setInput(SVDF::kActivationParam, &activation,
308 // Resets the state of SVDF op by filling it with 0's.
314 // Extracts the output tensor from the SVDF op.
339 SVDFOpModel svdf(/*batches=*/2, /*units=*/4, /*input_size=*/3,
341 svdf.SetWeightsFeature({-0.31930989, -0.36118156, 0.0079667, 0.37613347
[all...] |
| SVDF.cpp | 17 #include "SVDF.h"
39 SVDF::SVDF(const Operation& operation,
55 bool SVDF::Prepare(const Operation &operation,
66 GetInput(operation, operands, SVDF::kInputTensor);
68 GetInput(operation, operands, SVDF::kWeightsFeatureTensor);
70 GetInput(operation, operands, SVDF::kWeightsTimeTensor);
105 bool SVDF::Eval() {
|
| SVDF.h | 46 class SVDF {
48 SVDF(const android::hardware::neuralnetworks::V1_1::Operation &operation,
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| svdf_test.cc | 15 // Unit test for TFLite SVDF op.
128 // Derived class of SingleOpModel, which is used to test SVDF TFLite op.
169 // Resets the state of SVDF op by filling it with 0's.
178 // Extracts the output tensor from the SVDF op.
201 SVDFOpModel svdf(/*batches=*/2, /*units=*/4, /*input_size=*/3,
203 svdf.SetWeightsFeature({-0.31930989, -0.36118156, 0.0079667, 0.37613347,
207 svdf.SetWeightsTime(
220 svdf.ResetState();
221 const int svdf_num_batches = svdf.num_batches();
222 const int svdf_input_size = svdf.input_size()
[all...] |
| svdf.cc | 33 namespace svdf { namespace in namespace:tflite::ops::builtin
212 } // namespace svdf
215 static TfLiteRegistration r = {svdf::Init, svdf::Free, svdf::Prepare,
216 svdf::Eval};
|
| Android.bp | 72 "svdf.cc",
|
| /frameworks/ml/nn/runtime/test/generated/models/ |
| svdf.model.cpp | 1 // Generated file (from: svdf.mod.py). Do not edit
|
| /external/tensorflow/tensorflow/contrib/lite/toco/tensorflow_graph_matching/ |
| resolve_svdf.cc | 104 svdf_node->set_op("Svdf");
111 // Add the rest of the inputs to Svdf cell: weights and bias.
130 // Unexpected input for Svdf op.
131 LOG(FATAL) << "Unexpected input node for SVDF op! Accepted inputs are: "
138 // Add Svdf activation and rank.
262 if (node_name_to_upper.find("SVDF", 0) != string::npos) {
|
| resolve_svdf.h | 59 // Infer the value of Svdf filter rank, by looking up a reshape operator which
74 // and it creates an SVDF Op from them.
|
| resolve_svdf_test.cc | 180 // After CreateNodes in each cluster we have three nodes: Svdf,
194 // Checking the Svdf node attributes (rank and activation type) are
|
| /external/tensorflow/tensorflow/contrib/lite/models/testdata/g3doc/ |
| README.md | 28 size of 40 (float), an output size of 7 (float), one Svdf layer, and four fully
103 [Speech hotword model (Svdf
106 [Speech hotword model (Svdf
|
| /external/tensorflow/tensorflow/contrib/lite/ |
| tflite_static.bp | 67 "kernels/svdf.cc",
|
| /frameworks/ml/nn/runtime/test/generated/examples/ |
| svdf.example.cpp | 1 // Generated file (from: svdf.mod.py). Do not edit
|
| /frameworks/ml/nn/common/ |
| Android.bp | 74 "operations/SVDF.cpp",
|
| /frameworks/ml/nn/runtime/test/generated/vts_models/ |
| svdf.model.cpp | 90 .type = OperationType::SVDF,
|
| svdf2.model.cpp | 90 .type = OperationType::SVDF,
|
| svdf2_relaxed.model.cpp | 90 .type = OperationType::SVDF,
|
| svdf_relaxed.model.cpp | 90 .type = OperationType::SVDF,
|
| svdf_state.model.cpp | 90 .type = OperationType::SVDF,
|
| svdf_state_relaxed.model.cpp | 90 .type = OperationType::SVDF,
|
| /frameworks/ml/nn/runtime/test/specs/V1_0/ |
| svdf.mod.py | 36 model = model.Operation("SVDF", input, weights_feature, weights_time, bias, state_in,
|
| svdf_state.mod.py | 34 model = model.Operation("SVDF", input, weights_feature, weights_time, bias, state_in,
|
| /frameworks/ml/nn/runtime/test/specs/V1_1/ |
| svdf_relaxed.mod.py | 36 model = model.Operation("SVDF", input, weights_feature, weights_time, bias, state_in,
|
| svdf_state_relaxed.mod.py | 34 model = model.Operation("SVDF", input, weights_feature, weights_time, bias, state_in,
|
| /hardware/interfaces/neuralnetworks/1.0/vts/functional/ |
| Models.h | 179 FN(svdf) \
|
