Searched defs:params_size_in_bytes (Results 1 – 3 of 3) sorted by relevance
1172 int64_t params_size_in_bytes = rnn_desc->ParamsSizeInBytes(); in Compute() local1245 int64_t params_size_in_bytes = rnn_desc->ParamsSizeInBytes(); in Compute() local1422 int64_t params_size_in_bytes = rnn_desc->ParamsSizeInBytes(); in Compute() local
1202 int64_t params_size_in_bytes, ParamsRegions weights, in CudnnRnnParamsDescriptor()1218 int64_t params_size_in_bytes() const { return params_size_in_bytes_; } in params_size_in_bytes() function in stream_executor::gpu::__anon96353b0e0311::CudnnRnnParamsDescriptor1552 int64_t params_size_in_bytes = static_cast<int64_t>(params_size); in Create() local1836 size_t params_size_in_bytes = 0; in CheckRNNParameterSize() local
1793 int64_t params_size_in_bytes() const { return params_size_in_bytes_; } in params_size_in_bytes() function in stream_executor::gpu::MIOpenRnnParamsDescriptor2089 size_t params_size_in_bytes = 0; in CheckRNNParameterSize() local