Searched refs:weights_feature (Results 1 - 13 of 13) sorted by relevance

/frameworks/ml/nn/runtime/test/specs/V1_0/
H A Dsvdf_state.mod.py25 weights_feature = Input("weights_feature", "TENSOR_FLOAT32", "{%d, %d}" % (units, input_size)) variable
34 model = model.Operation("SVDF", input, weights_feature, weights_time, bias, state_in,
38 weights_feature: [
H A Dsvdf.mod.py27 weights_feature = Input("weights_feature", "TENSOR_FLOAT32", "{%d, %d}" % (features, input_size)) variable
36 model = model.Operation("SVDF", input, weights_feature, weights_time, bias, state_in,
41 weights_feature: [
H A Dsvdf2.mod.py27 weights_feature = Input("weights_feature", "TENSOR_FLOAT32", "{%d, %d}" % (features, input_size)) variable
36 model = model.Operation("SVDF", input, weights_feature, weights_time, bias, state_in,
41 weights_feature: [
/frameworks/ml/nn/runtime/test/specs/V1_1/
H A Dsvdf_state_relaxed.mod.py25 weights_feature = Input("weights_feature", "TENSOR_FLOAT32", "{%d, %d}" % (units, input_size)) variable
34 model = model.Operation("SVDF", input, weights_feature, weights_time, bias, state_in,
39 weights_feature: [
H A Dsvdf2_relaxed.mod.py27 weights_feature = Input("weights_feature", "TENSOR_FLOAT32", "{%d, %d}" % (features, input_size)) variable
36 model = model.Operation("SVDF", input, weights_feature, weights_time, bias, state_in,
42 weights_feature: [
H A Dsvdf_relaxed.mod.py27 weights_feature = Input("weights_feature", "TENSOR_FLOAT32", "{%d, %d}" % (features, input_size)) variable
36 model = model.Operation("SVDF", input, weights_feature, weights_time, bias, state_in,
42 weights_feature: [
/frameworks/ml/nn/runtime/test/generated/models/
H A Dsvdf.model.cpp12 auto weights_feature = model->addOperand(&type1); local
25 model->addOperation(ANEURALNETWORKS_SVDF, {input, weights_feature, weights_time, bias, state_in, rank_param, activation_param}, {state_out, output});
28 {input, weights_feature, weights_time, bias, state_in},
H A Dsvdf2.model.cpp12 auto weights_feature = model->addOperand(&type1); local
25 model->addOperation(ANEURALNETWORKS_SVDF, {input, weights_feature, weights_time, bias, state_in, rank_param, activation_param}, {state_out, output});
28 {input, weights_feature, weights_time, bias, state_in},
H A Dsvdf2_relaxed.model.cpp12 auto weights_feature = model->addOperand(&type1); local
25 model->addOperation(ANEURALNETWORKS_SVDF, {input, weights_feature, weights_time, bias, state_in, rank_param, activation_param}, {state_out, output});
28 {input, weights_feature, weights_time, bias, state_in},
H A Dsvdf_relaxed.model.cpp12 auto weights_feature = model->addOperand(&type1); local
25 model->addOperation(ANEURALNETWORKS_SVDF, {input, weights_feature, weights_time, bias, state_in, rank_param, activation_param}, {state_out, output});
28 {input, weights_feature, weights_time, bias, state_in},
H A Dsvdf_state.model.cpp12 auto weights_feature = model->addOperand(&type1); local
25 model->addOperation(ANEURALNETWORKS_SVDF, {input, weights_feature, weights_time, bias, state_in, rank_param, activation_param}, {state_out, output});
28 {input, weights_feature, weights_time, bias, state_in},
H A Dsvdf_state_relaxed.model.cpp12 auto weights_feature = model->addOperand(&type1); local
25 model->addOperation(ANEURALNETWORKS_SVDF, {input, weights_feature, weights_time, bias, state_in, rank_param, activation_param}, {state_out, output});
28 {input, weights_feature, weights_time, bias, state_in},
/frameworks/ml/nn/common/operations/
H A DSVDF.cpp67 const RunTimeOperandInfo *weights_feature = local
76 const uint32_t num_filters = SizeOfDimension(weights_feature, 0);
80 NN_CHECK_EQ(SizeOfDimension(input, 1), SizeOfDimension(weights_feature, 1));
115 // Compute conv1d(inputs, weights_feature).

Completed in 125 milliseconds