Searched refs:state_in (Results 1 - 12 of 12) sorted by relevance

/frameworks/ml/nn/runtime/test/specs/V1_0/
H A Dsvdf_state.mod.py28 state_in = Input("state_in", "TENSOR_FLOAT32", "{%d, %d}" % (batches, memory_size*units)) variable
34 model = model.Operation("SVDF", input, weights_feature, weights_time, bias, state_in,
63 input0[state_in] = [
H A Dsvdf.mod.py30 state_in = Input("state_in", "TENSOR_FLOAT32", "{%d, %d}" % (batches, memory_size*features)) variable
36 model = model.Operation("SVDF", input, weights_feature, weights_time, bias, state_in,
60 state_in: [0 for _ in range(batches * memory_size * features)],
H A Dsvdf2.mod.py30 state_in = Input("state_in", "TENSOR_FLOAT32", "{%d, %d}" % (batches, memory_size*features)) variable
36 model = model.Operation("SVDF", input, weights_feature, weights_time, bias, state_in,
75 state_in: [0 for _ in range(batches * memory_size * features)],
/frameworks/ml/nn/runtime/test/specs/V1_1/
H A Dsvdf_state_relaxed.mod.py28 state_in = Input("state_in", "TENSOR_FLOAT32", "{%d, %d}" % (batches, memory_size*units)) variable
34 model = model.Operation("SVDF", input, weights_feature, weights_time, bias, state_in,
64 input0[state_in] = [
H A Dsvdf2_relaxed.mod.py30 state_in = Input("state_in", "TENSOR_FLOAT32", "{%d, %d}" % (batches, memory_size*features)) variable
36 model = model.Operation("SVDF", input, weights_feature, weights_time, bias, state_in,
76 state_in: [0 for _ in range(batches * memory_size * features)],
H A Dsvdf_relaxed.mod.py30 state_in = Input("state_in", "TENSOR_FLOAT32", "{%d, %d}" % (batches, memory_size*features)) variable
36 model = model.Operation("SVDF", input, weights_feature, weights_time, bias, state_in,
61 state_in: [0 for _ in range(batches * memory_size * features)],
/frameworks/ml/nn/runtime/test/generated/models/
H A Dsvdf.model.cpp15 auto state_in = model->addOperand(&type4); 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.cpp15 auto state_in = model->addOperand(&type4); 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.cpp15 auto state_in = model->addOperand(&type4); 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.cpp15 auto state_in = model->addOperand(&type4); 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.cpp15 auto state_in = model->addOperand(&type4); 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.cpp15 auto state_in = model->addOperand(&type4); 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},

Completed in 2077 milliseconds