/external/tensorflow/tensorflow/python/keras/layers/ |
D | recurrent_v2.py | 78 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 …]
|
D | cudnn_recurrent.py | 61 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:
|
D | recurrent.py | 401 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/ |
D | rnn.py | 52 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/ |
D | unidirectional_sequence_rnn.cc | 82 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 …]
|
D | bidirectional_sequence_rnn.cc | 148 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 …]
|
D | unidirectional_sequence_rnn_test.cc | 174 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()
|
D | lstm_eval.h | 122 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,
|
D | bidirectional_sequence_lstm.cc | 436 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()
|
D | unidirectional_sequence_lstm_test.cc | 33 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 …]
|
D | lstm_eval.cc | 1711 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/ |
D | api_def_CudnnRNNV3.pbtxt | 19 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
|
D | api_def_CudnnRNNBackpropV3.pbtxt | 19 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/ |
D | cudnn_rnn_ops.cc | 566 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/ |
D | builtin_op_data.h | 151 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/ |
D | split-merged-operands.mlir | 7 …_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…
|
D | prepare-composite-functions-tf.mlir | 181 …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/ |
D | lstm.cc | 287 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/ |
D | rnn.py | 356 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:
|
D | ctc_ops.py | 81 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/ |
D | ctc_loss_op_test.py | 70 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/ |
D | unidirectional_sequence_rnn.mlir | 62 // 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/ |
D | tensorflow.lite.experimental.nn.pbtxt | 13 …nitial_state\', \'dtype\', \'parallel_iterations\', \'swap_memory\', \'time_major\', \'scope\'], v…
|
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
D | CudnnRNNV3.pbtxt | 250 name: "time_major" 389 name: "time_major"
|
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
D | lstm.cc | 355 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()
|