Home
last modified time | relevance | path

Searched defs:time_major (Results 1 – 25 of 34) sorted by relevance

12

/external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/
Dcudnn_rnn_ops_test.py72 time_major=True, argument
352 time_major, argument
388 variable_seq_lengths, time_major, dynamic_shape_input): argument
412 num_layers, variable_seq_lengths, time_major, argument
439 variable_seq_lengths, time_major, dynamic_shape_input): argument
471 num_layers, variable_seq_lengths, time_major, argument
508 num_layers, variable_seq_lengths, time_major, argument
560 time_major=True, argument
734 time_major, argument
770 variable_seq_lengths, time_major, dynamic_shape_input): argument
[all …]
/external/tensorflow/tensorflow/python/keras/layers/
Drecurrent_v2.py172 time_major=False, argument
322 recurrent_activation, time_major): argument
392 def cudnn_gru(inputs, init_h, kernel, recurrent_kernel, bias, time_major): argument
541 time_major=False, argument
715 activation, recurrent_activation, time_major): argument
786 time_major): argument
Dgru_v2_test.py267 def test_time_major_and_go_backward(self, time_major, go_backwards): argument
Dcudnn_recurrent.py61 time_major=False, argument
Dlstm_v2_test.py410 def test_time_major_and_go_backward(self, time_major, go_backwards): argument
/external/tensorflow/tensorflow/contrib/recurrent/python/kernel_tests/
Dfunctional_rnn_test.py64 def _CreateInputs(self, time_major=False): argument
107 time_major=None, argument
148 is_dynamic, time_major=None, is_bidirectional=False): argument
/external/tensorflow/tensorflow/lite/experimental/examples/lstm/
Drnn.py49 time_major=True, argument
288 time_major=False, argument
/external/tensorflow/tensorflow/lite/kernels/
Dunidirectional_sequence_rnn.cc70 const bool time_major = params->time_major; in Prepare() local
155 const bool time_major = params->time_major; in EvalFloat() local
210 const bool time_major = params->time_major; in EvalHybrid() local
Dunidirectional_sequence_rnn_test.cc175 int batches, int sequence_len, int units, int size, bool time_major, in UnidirectionalRNNOpModel()
251 int size, bool time_major, in HybridUnidirectionalRNNOpModel()
Dunidirectional_sequence_lstm.cc266 const bool time_major = params->time_major; in Prepare() local
441 const bool time_major = params->time_major; in Eval() local
Dbidirectional_sequence_rnn.cc131 const bool time_major = params->time_major; in Prepare() local
295 const bool time_major = params->time_major; in EvalFloat() local
429 const bool time_major = params->time_major; in EvalHybrid() local
Dunidirectional_sequence_lstm_test.cc35 int sequence_length, bool time_major, bool use_cifg, in UnidirectionalLSTMOpModel()
244 bool time_major, bool use_cifg, bool use_peephole, in HybridUnidirectionalLSTMOpModel()
344 bool time_major = true) { in VerifyGoldens()
/external/tensorflow/tensorflow/contrib/rnn/python/ops/
Drnn.py133 time_major=False, argument
/external/tensorflow/tensorflow/contrib/seq2seq/python/kernel_tests/
Ddecoder_v2_test.py38 def _testDecodeRNN(self, time_major, maximum_iterations=None): argument
Ddecoder_test.py37 def _testDynamicDecodeRNN(self, time_major, maximum_iterations=None): argument
Dbeam_search_decoder_test.py478 def _testDynamicDecodeRNN(self, time_major, has_attention, argument
592 def _testDynamicDecodeRNN(self, time_major, has_attention, argument
/external/tensorflow/tensorflow/contrib/recurrent/python/ops/
Dfunctional_rnn.py280 time_major=False, argument
329 time_major=False, argument
/external/tensorflow/tensorflow/contrib/cudnn_rnn/python/ops/
Dcudnn_rnn_ops.py959 time_major=True, argument
1050 time_major=True, argument
1106 time_major=True, argument
1162 time_major=True, argument
1219 time_major=True, argument
1267 time_major=True, argument
1597 time_major=True): argument
1736 time_major=True, argument
1832 time_major=True, argument
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc562 const CudnnModelTypes& model_types, bool time_major, in ExtractForwardInput()
639 const CudnnModelTypes& model_types, bool time_major, in ExtractForwardInput()
655 const absl::Span<const int>& seq_lengths, bool time_major) { in CreateForwardAndBackwardIODescriptors()
737 const Tensor* sequence_lengths, bool time_major, in DoForward()
800 Tensor* params_backprop, const Tensor* sequence_lengths, bool time_major, in DoBackward()
1274 bool time_major) { in ComputeAndReturnAlgorithm()
1610 bool time_major() { return time_major_; } in time_major() function in tensorflow::CudnnRNNForwardOpV3
1662 bool time_major) { in ComputeImpl()
1876 bool time_major() { return time_major_; } in time_major() function in tensorflow::CudnnRNNBackwardOpV3
/external/tensorflow/tensorflow/contrib/cudnn_rnn/python/layers/
Dcudnn_rnn.py381 time_major=True, argument
498 def _forward(self, inputs, h, c, opaque_params, sequence_lengths, time_major, argument
/external/tensorflow/tensorflow/python/ops/
Drnn.py358 swap_memory=False, time_major=False, scope=None): argument
493 time_major=False, scope=None): argument
/external/tensorflow/tensorflow/contrib/seq2seq/python/ops/
Dsampler.py192 def __init__(self, time_major=False): argument
290 time_major=False, argument
384 time_major=False, argument
Dhelper.py233 def __init__(self, inputs, sequence_length, time_major=False, name=None): argument
323 time_major=False, seed=None, scheduling_seed=None, name=None): argument
423 time_major=False, seed=None, next_inputs_fn=None, argument
/external/tensorflow/tensorflow/lite/c/
Dbuiltin_op_data.h113 bool time_major; member
118 bool time_major; member
213 bool time_major; member
228 bool time_major; member
/external/tensorflow/tensorflow/python/kernel_tests/
Dctc_loss_op_test.py66 time_major=True): argument

12