Home
last modified time | relevance | path

Searched defs:input_mode (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc274 int input_size, RnnInputMode* input_mode) { in ToRNNInputMode()
1074 RnnInputMode input_mode; in ExtractCudnnRNNParamsInfo() local
1097 const RnnInputMode& input_mode, in CreateRnnDescriptor()
1124 const RnnInputMode& input_mode, in GetCachedRnnDescriptor()
1519 RnnInputMode input_mode; in ComputeAndReturnAlgorithm() local
1568 const RnnInputMode& input_mode, in MaybeAutotune()
1675 const RnnInputMode& input_mode, const Tensor* input, in MaybeAutotune()
1920 RnnInputMode input_mode; in ComputeImpl() local
/external/toybox/kconfig/
Dconf.c29 } input_mode = ask_all; variable
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1039 cudnnRNNInputMode_t ToCudnnRnnInputMode(dnn::RnnInputMode input_mode) { in ToCudnnRnnInputMode()
1163 int batch_size, cudnnRNNInputMode_t input_mode, in CudnnRnnDescriptor()
1192 int cell_size, int batch_size, cudnnRNNInputMode_t input_mode, in Create()
1312 cudnnRNNInputMode_t input_mode() const { return input_mode_; } in input_mode() function in stream_executor::gpu::CudnnRnnDescriptor
1397 cudnnRNNInputMode_t input_mode; in CheckAndFetchProjectionWeights() local
2355 int batch_size, dnn::RnnInputMode input_mode, in createRnnDescriptor()
/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_pimpl.cc374 int batch_size, dnn::RnnInputMode input_mode, in createRnnDescriptor()
Ddnn.h2014 dnn::RnnInputMode input_mode, in createRnnDescriptor()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc1711 miopenRNNInputMode_t ToMIOpenRnnInputMode(dnn::RnnInputMode input_mode) { in ToMIOpenRnnInputMode()
1815 miopenRNNInputMode_t input_mode, in MIOpenRnnDescriptor()
1859 miopenRNNInputMode_t input_mode() const { return input_mode_; } in input_mode() function in stream_executor::gpu::MIOpenRnnDescriptor
2568 int batch_size, dnn::RnnInputMode input_mode, in createRnnDescriptor()