| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | recurrent_v2.py | 75 def __init__(self, time_major, go_backwards, layer_name): argument
368 time_major=False, argument
542 time_major, go_backwards, sequence_lengths, argument
625 def gpu_gru(inputs, init_h, kernel, recurrent_kernel, bias, mask, time_major, argument
712 mask, time_major, go_backwards, sequence_lengths, argument
762 mask, time_major, go_backwards, sequence_lengths, argument
1092 time_major=False, argument
1323 mask, time_major, go_backwards, sequence_lengths, argument
1408 time_major, go_backwards, sequence_lengths): argument
1533 recurrent_kernel, bias, mask, time_major, argument
[all …]
|
| D | cudnn_recurrent.py | 57 time_major=False, argument
|
| D | recurrent.py | 397 time_major=False, argument
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | unidirectional_sequence_rnn.cc | 82 const bool time_major = params->time_major; in Prepare() local
216 const bool time_major = params->time_major; in EvalFloat() local
276 const bool time_major = params->time_major; in EvalHybrid() local
|
| D | unidirectional_sequence_rnn_test.cc | 174 int batches, int sequence_len, int units, int size, bool time_major, in UnidirectionalRNNOpModel()
252 int size, bool time_major, in HybridUnidirectionalRNNOpModel()
|
| D | bidirectional_sequence_rnn.cc | 148 const bool time_major = params->time_major; in Prepare() local
387 const bool time_major = params->time_major; in EvalFloat() local
533 const bool time_major = params->time_major; in EvalHybrid() local
|
| D | unidirectional_sequence_lstm_test.cc | 37 bool time_major, bool use_cifg, bool use_peephole, in HybridUnidirectionalLSTMOpModel()
139 bool time_major = true) { in VerifyGoldens()
2219 bool time_major, bool use_cifg, bool use_peephole, in LayerNormUnidirectionalLSTMOpModel()
2494 bool time_major, bool use_cifg, bool use_peephole, in UnidirectionalSequenceLSTMIntegerOpModel()
|
| D | unidirectional_sequence_lstm.cc | 894 const bool time_major = params->time_major; in Prepare() local
1224 const bool time_major = params->time_major; in Eval() local
|
| D | bidirectional_sequence_lstm.cc | 436 const bool time_major = params->time_major; in Prepare() local
1156 const bool time_major = params->time_major; in Eval() local
|
| D | lstm_eval.cc | 1745 TfLiteStatus EvalFloat( in EvalFloat()
1932 TfLiteStatus EvalHybrid( in EvalHybrid()
2233 const TfLiteLSTMParams* params, bool forward_sequence, bool time_major, in EvalInteger8x8_16()
|
| D | bidirectional_sequence_rnn_test.cc | 667 AuxInputMode aux_input_mode, bool time_major, in BidirectionalRNNOpModel()
|
| D | bidirectional_sequence_lstm_test.cc | 37 bool quantize_weights, bool time_major, in BidirectionalLSTMOpModel()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 569 const CudnnModelTypes& model_types, bool time_major, in ExtractForwardInput()
683 const CudnnModelTypes& model_types, bool time_major, in ExtractForwardInput()
700 const absl::Span<const int> seq_lengths, bool time_major) { in CreateForwardAndBackwardIODescriptors()
798 const Tensor* sequence_lengths, bool time_major, in DoForward()
882 Tensor* params_backprop, const Tensor* sequence_lengths, bool time_major, in DoBackward()
992 bool time_major) { in ShouldUsePaddedIO()
1496 bool var_seq_lengths, bool time_major, in ComputeAndReturnAlgorithm()
1841 bool time_major() { return time_major_; } in time_major() function in tensorflow::CudnnRNNForwardOpV3
1901 bool time_major, int num_proj) { in ComputeImpl()
2122 bool time_major() { return time_major_; } in time_major() function in tensorflow::CudnnRNNBackwardOpV3
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | rnn.py | 355 time_major=False, argument
511 time_major=False, argument
|
| D | ctc_ops.py | 77 time_major=True, argument
200 time_major=True, argument
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | builtin_op_data.h | 152 bool time_major; member
160 bool time_major; member
285 bool time_major; member
303 bool time_major; member
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 264 TfLiteStatus EvalCalibration( in EvalCalibration()
582 bool time_major = true; in lstm_eval() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| D | ctc_loss_op_test.py | 67 time_major=True): argument
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| D | lstm.cc | 340 TfLiteStatus EvalFloat( in EvalFloat()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | stream_executor_pimpl.cc | 409 const absl::Span<const int>& seq_lengths, bool time_major, in createRnnSequenceTensorDescriptor()
|
| D | dnn.h | 2070 bool time_major, dnn::DataType data_type) { in createRnnSequenceTensorDescriptor()
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend.py | 4180 time_major=False, argument
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1672 const absl::Span<const int>& seq_lengths, bool time_major, in Create()
2450 const absl::Span<const int>& seq_lengths, bool time_major, in createRnnSequenceTensorDescriptor()
|