Searched refs:ParamsSizeInBytes (Results 1 – 4 of 4) sorted by relevance
1043 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()
173 virtual int64 ParamsSizeInBytes() const { return -1; }
1120 int64 ParamsSizeInBytes() const override { in ParamsSizeInBytes() function in stream_executor::gpu::CudnnRnnDescriptor1478 rnn_desc.ParamsSizeInBytes()) { in CheckRNNParameterSize()
1731 int64 ParamsSizeInBytes() const override { in ParamsSizeInBytes() function in stream_executor::gpu::MIOpenRnnDescriptor1964 rnn_desc.ParamsSizeInBytes(); in CheckRNNParameterSize()