Home
last modified time | relevance | path

Searched refs:params_size_in_bytes_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc1658 int64 params_size_in_bytes() const { return params_size_in_bytes_; } in params_size_in_bytes()
1672 int64 params_size_in_bytes_; member in stream_executor::gpu::MIOpenRnnParamsDescriptor
2235 : handle_(nullptr), rnn_desc_(&rnn_desc), params_size_in_bytes_(0) { in MIOpenRnnParamsDescriptor()
2253 params_size_in_bytes_ = static_cast<int64>(params_size); in MIOpenRnnParamsDescriptor()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc971 params_size_in_bytes_(params_size_in_bytes), in CudnnRnnParamsDescriptor()
984 int64 params_size_in_bytes() const { return params_size_in_bytes_; } in params_size_in_bytes()
994 int64 params_size_in_bytes_; member in stream_executor::gpu::__anonf41f84630311::CudnnRnnParamsDescriptor