Searched refs:ParamsWeightRegions (Results 1 – 4 of 4) sorted by relevance
1121 CHECK(num_params_ == rnn_desc->ParamsWeightRegions().size()) in Compute()1123 << rnn_desc->ParamsWeightRegions().size(); in Compute()1124 for (int i = 0; i < rnn_desc->ParamsWeightRegions().size(); i++) { in Compute()1125 int64 size_in_bytes = rnn_desc->ParamsWeightRegions()[i].size; in Compute()1160 input_ptr, rnn_desc->ParamsWeightRegions()[i].offset, size_in_bytes); in Compute()1227 RestoreParams<T>(weights, rnn_desc->ParamsWeightRegions(), &output_ptr, in Compute()
174 virtual ParamsRegions ParamsWeightRegions() const { return ParamsRegions(); }
1126 ParamsRegions ParamsWeightRegions() const override { in ParamsWeightRegions() function in stream_executor::gpu::CudnnRnnDescriptor
1738 ParamsRegions ParamsWeightRegions() const override { in ParamsWeightRegions() function in stream_executor::gpu::MIOpenRnnDescriptor