Home
last modified time | relevance | path

Searched refs:ParamsSizeInBytes (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc1043 int64 params_size_in_bytes = rnn_desc->ParamsSizeInBytes(); in Compute()
1087 int64 params_size_in_bytes = rnn_desc->ParamsSizeInBytes(); in Compute()
1215 int64 params_size_in_bytes = rnn_desc->ParamsSizeInBytes(); in Compute()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h173 virtual int64 ParamsSizeInBytes() const { return -1; }
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1120 int64 ParamsSizeInBytes() const override { in ParamsSizeInBytes() function in stream_executor::gpu::CudnnRnnDescriptor
1478 rnn_desc.ParamsSizeInBytes()) { in CheckRNNParameterSize()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc1731 int64 ParamsSizeInBytes() const override { in ParamsSizeInBytes() function in stream_executor::gpu::MIOpenRnnDescriptor
1964 rnn_desc.ParamsSizeInBytes(); in CheckRNNParameterSize()