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.cc272 int input_size, RnnInputMode* input_mode) { in ToRNNInputMode()
1033 RnnInputMode input_mode; in ExtractCudnnRNNParamsInfo() local
1056 const RnnInputMode& input_mode, in CreateRnnDescriptor()
1083 const RnnInputMode& input_mode, in GetCachedRnnDescriptor()
1478 RnnInputMode input_mode; in ComputeAndReturnAlgorithm() local
1527 const RnnInputMode& input_mode, in MaybeAutoTune()
1634 const RnnInputMode& input_mode, const Tensor* input, in MaybeAutoTune()
1878 RnnInputMode input_mode; in ComputeImpl() local
/external/toybox/kconfig/
Dconf.c29 } input_mode = ask_all; variable
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc938 cudnnRNNInputMode_t ToCudnnRnnInputMode(dnn::RnnInputMode input_mode) { in ToCudnnRnnInputMode()
1061 int batch_size, cudnnRNNInputMode_t input_mode, in CudnnRnnDescriptor()
1090 int cell_size, int batch_size, cudnnRNNInputMode_t input_mode, in Create()
1207 cudnnRNNInputMode_t input_mode() const { return input_mode_; } in input_mode() function in stream_executor::gpu::CudnnRnnDescriptor
1292 cudnnRNNInputMode_t input_mode; in CheckAndFetchProjectionWeights() local
2072 int batch_size, dnn::RnnInputMode input_mode, in createRnnDescriptor()
/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_pimpl.cc367 int batch_size, dnn::RnnInputMode input_mode, in createRnnDescriptor()
Ddnn.h2112 dnn::RnnInputMode input_mode, in createRnnDescriptor()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc1660 miopenRNNInputMode_t ToMIOpenRnnInputMode(dnn::RnnInputMode input_mode) { in ToMIOpenRnnInputMode()
1764 miopenRNNInputMode_t input_mode, in MIOpenRnnDescriptor()
1808 miopenRNNInputMode_t input_mode() const { return input_mode_; } in input_mode() function in stream_executor::gpu::MIOpenRnnDescriptor
2517 int batch_size, dnn::RnnInputMode input_mode, in createRnnDescriptor()