Home
last modified time | relevance | path

Searched refs:time_major (Results 1 – 25 of 74) sorted by relevance

123

/external/tensorflow/tensorflow/python/keras/layers/
Drecurrent_v2.py78 def __init__(self, time_major, go_backwards, layer_name): argument
79 self.time_major = time_major
87 'time_major': self.time_major,
103 self.time_major, self.go_backwards, self.layer_name)
371 time_major=False, argument
401 time_major=time_major,
419 self._defun_wrapper = _DefunWrapper(time_major, go_backwards, 'gru')
435 timesteps = input_shape[0] if self.time_major else input_shape[1]
455 time_major=self.time_major,
498 'time_major': self.time_major,
[all …]
Dcudnn_recurrent.py61 time_major=False, argument
70 self.time_major = time_major
130 'time_major': self.time_major,
273 if not self.time_major:
311 if self.time_major:
469 if not self.time_major:
513 if self.time_major:
Drecurrent.py401 time_major=False, argument
430 self.time_major = time_major
485 if self.time_major:
496 if self.time_major:
551 batch_index, time_step_index = (1, 0) if self.time_major else (0, 1)
561 return shape[1:] if self.time_major else (shape[0],) + shape[2:]
651 batch_size = input_shape[1] if self.time_major else input_shape[0]
761 timesteps = input_shape[0] if self.time_major else input_shape[1]
815 time_major=self.time_major,
924 batch_size = spec_shape[1] if self.time_major else spec_shape[0]
[all …]
/external/tensorflow/tensorflow/lite/experimental/examples/lstm/
Drnn.py52 time_major=True, argument
164 assert time_major
215 if not time_major:
274 if not time_major:
291 time_major=False, argument
381 time_major=time_major,
385 if not time_major:
420 time_major=time_major,
/external/tensorflow/tensorflow/lite/kernels/
Dunidirectional_sequence_rnn.cc82 const bool time_major = params->time_major; in Prepare() local
84 (time_major) ? input->dims->data[1] : input->dims->data[0]; in Prepare()
86 (time_major) ? input->dims->data[0] : input->dims->data[1]; in Prepare()
108 output_size_array->data[0] = (time_major) ? max_time : batch_size; in Prepare()
109 output_size_array->data[1] = (time_major) ? batch_size : max_time; in Prepare()
216 const bool time_major = params->time_major; in EvalFloat() local
218 (time_major) ? input->dims->data[1] : input->dims->data[0]; in EvalFloat()
220 (time_major) ? input->dims->data[0] : input->dims->data[1]; in EvalFloat()
228 if (time_major) { in EvalFloat()
276 const bool time_major = params->time_major; in EvalHybrid() local
[all …]
Dbidirectional_sequence_rnn.cc148 const bool time_major = params->time_major; in Prepare() local
150 (time_major) ? input->dims->data[1] : input->dims->data[0]; in Prepare()
152 (time_major) ? input->dims->data[0] : input->dims->data[1]; in Prepare()
351 fw_output_size_array->data[0] = (time_major) ? max_time : batch_size; in Prepare()
352 fw_output_size_array->data[1] = (time_major) ? batch_size : max_time; in Prepare()
385 const bool time_major = params->time_major; in EvalFloat() local
387 (time_major) ? input->dims->data[1] : input->dims->data[0]; in EvalFloat()
389 (time_major) ? input->dims->data[0] : input->dims->data[1]; in EvalFloat()
418 if (time_major) { in EvalFloat()
531 const bool time_major = params->time_major; in EvalHybrid() local
[all …]
Dunidirectional_sequence_rnn_test.cc174 int batches, int sequence_len, int units, int size, bool time_major, in UnidirectionalRNNOpModel() argument
190 CreateSequenceRNNOptions(builder_, time_major, in UnidirectionalRNNOpModel()
194 if (time_major) { in UnidirectionalRNNOpModel()
252 int size, bool time_major, in HybridUnidirectionalRNNOpModel() argument
255 : UnidirectionalRNNOpModel(batches, sequence_len, units, size, time_major, in HybridUnidirectionalRNNOpModel()
Dlstm_eval.h122 const TfLiteLSTMParams* params, bool forward_sequence, bool time_major,
160 bool forward_sequence, bool time_major, int output_offset,
191 const TfLiteLSTMParams* params, bool forward_sequence, bool time_major,
Dbidirectional_sequence_lstm.cc436 const bool time_major = params->time_major; in Prepare() local
437 const int max_time = time_major ? input->dims->data[0] : input->dims->data[1]; in Prepare()
438 const int n_batch = time_major ? input->dims->data[1] : input->dims->data[0]; in Prepare()
551 fw_output_size->data[0] = time_major ? max_time : n_batch; in Prepare()
552 fw_output_size->data[1] = time_major ? n_batch : max_time; in Prepare()
616 bw_output_size->data[0] = time_major ? max_time : n_batch; in Prepare()
617 bw_output_size->data[1] = time_major ? n_batch : max_time; in Prepare()
1152 const bool time_major = params->time_major; in Eval() local
1191 /*forward_sequence=*/true, time_major, /*output_offset=*/0, in Eval()
1211 /*forward_sequence=*/false, time_major, bw_output_offset, in Eval()
Dunidirectional_sequence_lstm_test.cc33 int sequence_length, bool time_major, bool use_cifg, in UnidirectionalLSTMOpModel() argument
131 proj_clip, time_major, asymmetric_quantize_inputs) in UnidirectionalLSTMOpModel()
289 bool time_major, bool use_cifg, bool use_peephole, in HybridUnidirectionalLSTMOpModel() argument
294 n_batch, n_input, n_cell, n_output, sequence_length, time_major, in HybridUnidirectionalLSTMOpModel()
391 bool time_major = true) { in VerifyGoldens() argument
398 if (time_major) { in VerifyGoldens()
425 if (time_major) { in VerifyGoldens()
2471 bool time_major, bool use_cifg, bool use_peephole, in LayerNormUnidirectionalLSTMOpModel() argument
2476 n_batch, n_input, n_cell, n_output, sequence_length, time_major, in LayerNormUnidirectionalLSTMOpModel()
2746 bool time_major, bool use_cifg, bool use_peephole, in UnidirectionalSequenceLSTMIntegerOpModel() argument
[all …]
Dlstm_eval.cc1711 const TfLiteLSTMParams* params, bool forward_sequence, bool time_major, in EvalFloat()
1717 max_time = (time_major) ? input->dims->data[0] : input->dims->data[1]; in EvalFloat()
1718 n_batch = (time_major) ? input->dims->data[1] : input->dims->data[0]; in EvalFloat()
1754 if (time_major) { in EvalFloat()
1902 bool forward_sequence, bool time_major, int output_offset, in EvalHybrid()
1920 max_time = (time_major) ? input->dims->data[0] : input->dims->data[1]; in EvalHybrid()
1921 n_batch = (time_major) ? input->dims->data[1] : input->dims->data[0]; in EvalHybrid()
1963 if (time_major) { in EvalHybrid()
2169 const TfLiteLSTMParams* params, bool forward_sequence, bool time_major, in EvalInteger8x8_16() argument
2182 max_time = (time_major) ? input->dims->data[0] : input->dims->data[1]; in EvalInteger8x8_16()
[all …]
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_CudnnRNNV3.pbtxt19 input: If time_major is true, this is a 3-D tensor with the shape of
20 [seq_length, batch_size, input_size]. If time_major is false, the shape is
22 input_h: If time_major is true, this is a 3-D tensor with the shape of
23 [num_layer * dir, batch_size, num_units]. If time_major is false, the shape
32 output: If time_major is true, this is a 3-D tensor with the shape of
33 [seq_length, batch_size, dir * num_units]. If time_major is false, the
39 time_major: Indicates whether the input/output format is time major or batch
Dapi_def_CudnnRNNBackpropV3.pbtxt19 input: If time_major is true, this is a 3-D tensor with the shape of
20 [seq_length, batch_size, input_size]. If time_major is false, the shape is
22 input_h: If time_major is true, this is a 3-D tensor with the shape of
23 [num_layer * dir, batch_size, num_units]. If time_major is false, the shape
32 output: If time_major is true, this is a 3-D tensor with the shape of
33 [seq_length, batch_size, dir * num_units]. If time_major is false, the
42 time_major: Indicates whether the input/output format is time major or batch
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc566 const CudnnModelTypes& model_types, bool time_major, in ExtractForwardInput() argument
581 if (time_major) { in ExtractForwardInput()
598 if (time_major) { in ExtractForwardInput()
607 if (time_major) { in ExtractForwardInput()
624 if (time_major) { in ExtractForwardInput()
654 if (time_major) { in ExtractForwardInput()
666 if (time_major) { in ExtractForwardInput()
680 const CudnnModelTypes& model_types, bool time_major, in ExtractForwardInput() argument
686 return ExtractForwardInput(context, model_types, time_major, input, input_h, in ExtractForwardInput()
697 const absl::Span<const int>& seq_lengths, bool time_major) { in CreateForwardAndBackwardIODescriptors() argument
[all …]
/external/tensorflow/tensorflow/lite/c/
Dbuiltin_op_data.h151 bool time_major; member
159 bool time_major; member
284 bool time_major; member
302 bool time_major; member
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Dsplit-merged-operands.mlir7 …_1]], %arg0, %arg0, %arg0, %arg0) {fused_activation_function = "NONE", time_major = true} : (tenso…
10 …, %0, %arg0, %arg0, %arg0, %arg0) {fused_activation_function = "NONE", time_major = true} : (tenso…
18 …_1]], %arg0, %arg0, %arg0, %arg0) {fused_activation_function = "NONE", time_major = true} : (tenso…
21 …_3]], %arg0, %arg0, %arg0, %arg0) {fused_activation_function = "NONE", time_major = true} : (tenso…
24 …, %0, %arg0, %arg0, %arg0, %arg0) {fused_activation_function = "NONE", time_major = true} : (tenso…
25 …, %0, %arg0, %arg0, %arg0, %arg0) {fused_activation_function = "NONE", time_major = true} : (tenso…
Dprepare-composite-functions-tf.mlir181 …f2-8496419a608c", tf.api_preferred_device = "CPU", tf.go_backwards = false, tf.time_major = true} {
192 …f2-8496419a608c", tf.api_preferred_device = "CPU", tf.go_backwards = false, tf.time_major = true} {
207 …2, fused_activation_function = "TANH", proj_clip = 0.000000e+00 : f32, time_major = true} : (tenso…
222 …2-8496419a608c", tf.api_preferred_device = "CPU", tf.go_backwards = false, tf.time_major = false} {
233 …2-8496419a608c", tf.api_preferred_device = "CPU", tf.go_backwards = false, tf.time_major = false} {
248 …2, fused_activation_function = "TANH", proj_clip = 0.000000e+00 : f32, time_major = false} : (tens…
264 …ef2-8496419a608c", tf.api_preferred_device = "CPU", tf.go_backwards = true, tf.time_major = true} {
275 …ef2-8496419a608c", tf.api_preferred_device = "CPU", tf.go_backwards = true, tf.time_major = true} {
292 …2, fused_activation_function = "TANH", proj_clip = 0.000000e+00 : f32, time_major = true} : (tenso…
308 …f2-8496419a608c", tf.api_preferred_device = "CPU", tf.go_backwards = true, tf.time_major = false} {
[all …]
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
Dlstm.cc287 const TfLiteLSTMParams* params, bool forward_sequence, bool time_major, in EvalCalibration()
295 max_time = (time_major) ? input->dims->data[0] : input->dims->data[1]; in EvalCalibration()
296 n_batch = (time_major) ? input->dims->data[1] : input->dims->data[0]; in EvalCalibration()
332 if (time_major) { in EvalCalibration()
581 bool time_major = true; in lstm_eval() local
585 time_major = true; in lstm_eval()
598 time_major = params->time_major; in lstm_eval()
623 /*time_major=*/time_major, in lstm_eval()
/external/tensorflow/tensorflow/python/ops/
Drnn.py356 time_major=False, argument
446 time_major=time_major,
450 if not time_major:
485 time_major=time_major,
512 time_major=False, argument
637 if not time_major:
696 if not time_major:
Dctc_ops.py81 time_major=True, argument
193 time_major,
204 time_major=True, argument
219 if not time_major:
711 time_major=logits_time_major,
729 time_major=logits_time_major,
817 time_major=logits_time_major)
/external/tensorflow/tensorflow/python/kernel_tests/
Dctc_loss_op_test.py70 time_major=True): argument
81 logits_time_major=time_major)
273 time_major=False)
404 time_major=True)
458 time_major=True)
501 time_major=True)
561 time_major=True)
628 time_major=True)
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/mlir2flatbuffer/
Dunidirectional_sequence_rnn.mlir62 // CHECK-NEXT: time_major: true,
96 …"(%arg0, %arg1, %arg2, %arg3, %0) {fused_activation_function = "TANH", time_major = true} : (tenso…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.lite.experimental.nn.pbtxt13 …nitial_state\', \'dtype\', \'parallel_iterations\', \'swap_memory\', \'time_major\', \'scope\'], v…
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DCudnnRNNV3.pbtxt250 name: "time_major"
389 name: "time_major"
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
Dlstm.cc355 const TfLiteLSTMParams* params, bool forward_sequence, bool time_major, in EvalFloat()
364 max_time = (time_major) ? input->dims->data[0] : input->dims->data[1]; in EvalFloat()
365 n_batch = (time_major) ? input->dims->data[1] : input->dims->data[0]; in EvalFloat()
401 if (time_major) { in EvalFloat()

123