Searched defs:input_mode (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 275 int input_size, RnnInputMode* input_mode) { in ToRNNInputMode()
1075 RnnInputMode input_mode; in ExtractCudnnRNNParamsInfo() local
1098 const RnnInputMode& input_mode, in CreateRnnDescriptor()
1125 const RnnInputMode& input_mode, in GetCachedRnnDescriptor()
1520 RnnInputMode input_mode; in ComputeAndReturnAlgorithm() local
1570 const RnnInputMode& input_mode, in MaybeAutotune()
1677 const RnnInputMode& input_mode, const Tensor* input, in MaybeAutotune()
1922 RnnInputMode input_mode; in ComputeImpl() local
|
| /external/toybox/kconfig/ |
| D | conf.c | 29 } input_mode = ask_all; variable
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1112 cudnnRNNInputMode_t ToCudnnRnnInputMode(dnn::RnnInputMode input_mode) { in ToCudnnRnnInputMode()
1236 int batch_size, cudnnRNNInputMode_t input_mode, in CudnnRnnDescriptor()
1265 int cell_size, int batch_size, cudnnRNNInputMode_t input_mode, in Create()
1385 cudnnRNNInputMode_t input_mode() const { return input_mode_; } in input_mode() function in stream_executor::gpu::CudnnRnnDescriptor
1470 cudnnRNNInputMode_t input_mode; in CheckAndFetchProjectionWeights() local
2415 int batch_size, dnn::RnnInputMode input_mode, in createRnnDescriptor()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | stream_executor_pimpl.cc | 377 int batch_size, dnn::RnnInputMode input_mode, in createRnnDescriptor()
|
| D | dnn.h | 2040 dnn::RnnInputMode input_mode, in createRnnDescriptor()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1714 miopenRNNInputMode_t ToMIOpenRnnInputMode(dnn::RnnInputMode input_mode) { in ToMIOpenRnnInputMode()
1818 miopenRNNInputMode_t input_mode, in MIOpenRnnDescriptor()
1862 miopenRNNInputMode_t input_mode() const { return input_mode_; } in input_mode() function in stream_executor::gpu::MIOpenRnnDescriptor
2571 int batch_size, dnn::RnnInputMode input_mode, in createRnnDescriptor()
|