Searched refs:ParamsSizeInBytes (Results 1 – 4 of 4) sorted by relevance
1172 int64_t params_size_in_bytes = rnn_desc->ParamsSizeInBytes(); in Compute()1245 int64_t params_size_in_bytes = rnn_desc->ParamsSizeInBytes(); in Compute()1422 int64_t params_size_in_bytes = rnn_desc->ParamsSizeInBytes(); in Compute()
1393 int64_t ParamsSizeInBytes() const override { in ParamsSizeInBytes() function in stream_executor::gpu::CudnnRnnDescriptor1848 rnn_desc.ParamsSizeInBytes()) { in CheckRNNParameterSize()2012 /*weightSpaceSize=*/rnn_desc.ParamsSizeInBytes(), in DoRnnForwardImpl()2222 /*weightSpaceSize=*/rnn_desc.ParamsSizeInBytes(), in DoRnnBackwardImpl()2243 /*weightSpaceSize=*/rnn_desc.ParamsSizeInBytes(), in DoRnnBackwardImpl()
173 virtual int64_t ParamsSizeInBytes() const { return -1; } in ParamsSizeInBytes() function
1866 int64_t ParamsSizeInBytes() const override { in ParamsSizeInBytes() function in stream_executor::gpu::MIOpenRnnDescriptor2099 rnn_desc.ParamsSizeInBytes(); in CheckRNNParameterSize()