Searched defs:params_size_in_bytes (Results 1 – 3 of 3) sorted by relevance
1043 int64 params_size_in_bytes = rnn_desc->ParamsSizeInBytes(); in Compute() local1087 int64 params_size_in_bytes = rnn_desc->ParamsSizeInBytes(); in Compute() local1215 int64 params_size_in_bytes = rnn_desc->ParamsSizeInBytes(); in Compute() local
968 CudnnRnnParamsDescriptor(FilterDescriptor handle, int64 params_size_in_bytes, in CudnnRnnParamsDescriptor()984 int64 params_size_in_bytes() const { return params_size_in_bytes_; } in params_size_in_bytes() function in stream_executor::gpu::__anonf41f84630311::CudnnRnnParamsDescriptor1175 int64 params_size_in_bytes = static_cast<int64>(params_size); in Create() local1472 size_t params_size_in_bytes = 0; in CheckRNNParameterSize() local
1658 int64 params_size_in_bytes() const { return params_size_in_bytes_; } in params_size_in_bytes() function in stream_executor::gpu::MIOpenRnnParamsDescriptor1954 size_t params_size_in_bytes = 0; in CheckRNNParameterSize() local