| /frameworks/ml/nn/runtime/test/specs/V1_2/ |
| D | unidirectional_sequence_rnn.mod.py | 20 activation, time_major, output, input_data, weights_data, argument
23 time_major = Int32Scalar("time_major", time_major)
26 time_major).To(output)
153 time_major=0,
175 time_major=1,
|
| D | bidirectional_sequence_rnn.mod.py | 41 time_major, merge_outputs, fw_output, bw_output, input_data, argument
48 time_major = BoolScalar("time_major", time_major)
54 time_major, merge_outputs_scalar)
239 time_major=0,
289 time_major=1,
340 time_major=1,
396 time_major=0,
450 time_major=0,
512 time_major=0,
|
| D | bidirectional_sequence_lstm_merge_outputs.mod.py | 213 time_major = BoolScalar("time_major", True)
265 activation, cell_clip, proj_clip, merge_outputs, time_major,
|
| D | bidirectional_sequence_lstm_float16_batch_major_merge_outputs.mod.py | 214 time_major = BoolScalar("time_major", False)
266 activation, cell_clip, proj_clip, merge_outputs, time_major,
|
| D | bidirectional_sequence_lstm_cifg_peephole.mod.py | 214 time_major = BoolScalar("time_major", True)
266 activation, cell_clip, proj_clip, merge_outputs, time_major,
|
| D | bidirectional_sequence_lstm_float16_batch_major.mod.py | 214 time_major = BoolScalar("time_major", False)
266 activation, cell_clip, proj_clip, merge_outputs, time_major,
|
| D | bidirectional_sequence_lstm.mod.py | 214 time_major = BoolScalar("time_major", True)
266 activation, cell_clip, proj_clip, merge_outputs, time_major,
|
| D | bidirectional_sequence_lstm_aux_input.mod.py | 216 time_major = BoolScalar("time_major", True)
268 activation, cell_clip, proj_clip, merge_outputs, time_major,
|
| D | bidirectional_sequence_lstm_norm_fw_output.mod.py | 215 time_major = BoolScalar("time_major", True)
267 activation, cell_clip, proj_clip, merge_outputs, time_major,
|
| D | bidirectional_sequence_lstm_float16_batch_major_aux_input.mod.py | 217 time_major = BoolScalar("time_major", False)
269 activation, cell_clip, proj_clip, merge_outputs, time_major,
|
| /frameworks/ml/nn/runtime/test/specs/V1_3/ |
| D | unidirectional_sequence_rnn.mod.py | 20 activation, time_major, output, output_state, input_data, weights_data, argument
24 time_major = Int32Scalar("time_major", time_major)
27 time_major).To(output, output_state)
193 time_major=0,
219 time_major=1,
|
| D | bidirectional_sequence_rnn_1_3.mod.py | 41 time_major, merge_outputs, fw_output, bw_output, input_data, argument
48 time_major = BoolScalar("time_major", time_major)
54 time_major, merge_outputs_scalar)
245 time_major=0,
295 time_major=1,
347 time_major=1,
|
| D | bidirectional_sequence_rnn_state_output.mod.py | 44 fw_aux_weights, bw_aux_weights, activation, time_major, merge_outputs, argument
52 time_major = BoolScalar("time_major", time_major)
58 bw_aux_weights, activation, time_major,
242 time_major=0,
310 time_major=1,
357 time_major=1,
420 time_major=0,
485 time_major=0,
555 time_major=0,
|
| D | bidirectional_sequence_lstm_state_output.mod.py | 230 time_major = BoolScalar("time_major", True)
286 time_major,
|
| D | bidirectional_sequence_lstm.mod.py | 242 time_major = BoolScalar("time_major", True)
298 time_major,
|
| /frameworks/ml/nn/common/operations/ |
| D | BidirectionalSequenceLSTM.cpp | 189 params_.time_major = getScalarDataWithDefault<bool>(timeMajorOperand, false); in BidirectionalSequenceLSTM()
269 const uint32_t max_time = SizeOfDimension(input_, params_.time_major ? 0 : 1); in Prepare()
270 const uint32_t n_batch = SizeOfDimension(input_, params_.time_major ? 1 : 0); in Prepare()
388 fwOutputShape->dimensions[0] = params_.time_major ? max_time : n_batch; in Prepare()
389 fwOutputShape->dimensions[1] = params_.time_major ? n_batch : max_time; in Prepare()
414 bwOutputShape->dimensions[0] = params_.time_major ? max_time : n_batch; in Prepare()
415 bwOutputShape->dimensions[1] = params_.time_major ? n_batch : max_time; in Prepare()
517 GetBuffer<float>(fw_output_), fw_scratch_buffer.data(), params_.time_major, in Eval()
571 bw_scratch_buffer.data(), params_.time_major, kBackwardSequence); in Eval()
641 GetBuffer<_Float16>(fw_output_), fw_scratch_buffer.data(), params_.time_major, in Eval()
[all …]
|
| D | LSTM.h | 42 bool time_major; member
|