Home
last modified time | relevance | path

Searched refs:time_major1 (Results 1 – 2 of 2) sorted by relevance

/frameworks/ml/nn/runtime/test/generated/models/
Dunidirectional_sequence_rnn.model.cpp232 auto time_major1 = model->addOperand(&type6); in CreateModel_2() local
238 model->setOperandValue(time_major1, time_major1_init, sizeof(int32_t) * 1); in CreateModel_2()
239 …{input1, weights1, recurrent_weights1, bias1, hidden_state1, activation1, time_major1}, {output1}); in CreateModel_2()
267 auto time_major1 = model->addOperand(&type6); in CreateModel_relaxed_2() local
273 model->setOperandValue(time_major1, time_major1_init, sizeof(int32_t) * 1); in CreateModel_relaxed_2()
274 …{input1, weights1, recurrent_weights1, bias1, hidden_state1, activation1, time_major1}, {output1}); in CreateModel_relaxed_2()
304 auto time_major1 = model->addOperand(&type6); in CreateModel_float16_2() local
310 model->setOperandValue(time_major1, time_major1_init, sizeof(int32_t) * 1); in CreateModel_float16_2()
311 …{input1, weights1, recurrent_weights1, bias1, hidden_state1, activation1, time_major1}, {output1}); in CreateModel_float16_2()
339 auto time_major1 = model->addOperand(&type6); in CreateModel_dynamic_output_shape_2() local
[all …]
Dbidirectional_sequence_rnn.model.cpp319 auto time_major1 = model->addOperand(&type8); in CreateModel_2() local
327 model->setOperandValue(time_major1, time_major1_init, sizeof(bool8) * 1); in CreateModel_2()
330 …den_state1, aux_input1, fw_aux_weights1, bw_aux_weights1, activation1, time_major1, merge_outputs1… in CreateModel_2()
367 auto time_major1 = model->addOperand(&type8); in CreateModel_relaxed_2() local
375 model->setOperandValue(time_major1, time_major1_init, sizeof(bool8) * 1); in CreateModel_relaxed_2()
378 …den_state1, aux_input1, fw_aux_weights1, bw_aux_weights1, activation1, time_major1, merge_outputs1… in CreateModel_relaxed_2()
417 auto time_major1 = model->addOperand(&type8); in CreateModel_float16_2() local
425 model->setOperandValue(time_major1, time_major1_init, sizeof(bool8) * 1); in CreateModel_float16_2()
428 …den_state1, aux_input1, fw_aux_weights1, bw_aux_weights1, activation1, time_major1, merge_outputs1… in CreateModel_float16_2()
465 auto time_major1 = model->addOperand(&type8); in CreateModel_dynamic_output_shape_2() local
[all …]