Searched refs:ParamsSizeInBytes (Results 1 – 4 of 4) sorted by relevance
1130 int64 params_size_in_bytes = rnn_desc->ParamsSizeInBytes(); in Compute()1203 int64 params_size_in_bytes = rnn_desc->ParamsSizeInBytes(); in Compute()1380 int64 params_size_in_bytes = rnn_desc->ParamsSizeInBytes(); in Compute()
165 virtual int64 ParamsSizeInBytes() const { return -1; } in ParamsSizeInBytes() function
1215 int64 ParamsSizeInBytes() const override { in ParamsSizeInBytes() function in stream_executor::gpu::CudnnRnnDescriptor1667 rnn_desc.ParamsSizeInBytes()) { in CheckRNNParameterSize()
1812 int64 ParamsSizeInBytes() const override { in ParamsSizeInBytes() function in stream_executor::gpu::MIOpenRnnDescriptor2045 rnn_desc.ParamsSizeInBytes(); in CheckRNNParameterSize()