Searched defs:params_size_in_bytes (Results 1 – 3 of 3) sorted by relevance
1130 int64 params_size_in_bytes = rnn_desc->ParamsSizeInBytes(); in Compute() local1203 int64 params_size_in_bytes = rnn_desc->ParamsSizeInBytes(); in Compute() local1380 int64 params_size_in_bytes = rnn_desc->ParamsSizeInBytes(); in Compute() local
1027 CudnnRnnParamsDescriptor(FilterDescriptor handle, int64 params_size_in_bytes, in CudnnRnnParamsDescriptor()1043 int64 params_size_in_bytes() const { return params_size_in_bytes_; } in params_size_in_bytes() function in stream_executor::gpu::__anon423856410311::CudnnRnnParamsDescriptor1374 int64 params_size_in_bytes = static_cast<int64>(params_size); in Create() local1661 size_t params_size_in_bytes = 0; in CheckRNNParameterSize() local
1739 int64 params_size_in_bytes() const { return params_size_in_bytes_; } in params_size_in_bytes() function in stream_executor::gpu::MIOpenRnnParamsDescriptor2035 size_t params_size_in_bytes = 0; in CheckRNNParameterSize() local