Searched refs:ParamsRegions (Results 1 – 3 of 3) sorted by relevance
966 typedef dnn::RnnDescriptor::ParamsRegions ParamsRegions; typedef in stream_executor::gpu::__anonf41f84630311::CudnnRnnParamsDescriptor969 ParamsRegions weights, ParamsRegions biases) in CudnnRnnParamsDescriptor()985 ParamsRegions params_weights() const { in params_weights()988 ParamsRegions params_biases() const { in params_biases()995 ParamsRegions weights_;996 ParamsRegions biases_;1126 ParamsRegions ParamsWeightRegions() const override { in ParamsWeightRegions()1129 ParamsRegions ParamsBiasRegions() const override { in ParamsBiasRegions()1205 ParamsRegions weights; in Create()1206 ParamsRegions biases; in Create()
171 typedef std::vector<ParamsRegion> ParamsRegions;174 virtual ParamsRegions ParamsWeightRegions() const { return ParamsRegions(); }175 virtual ParamsRegions ParamsBiasRegions() const { return ParamsRegions(); }
1647 typedef dnn::RnnDescriptor::ParamsRegions ParamsRegions; typedef in stream_executor::gpu::MIOpenRnnParamsDescriptor1659 ParamsRegions params_weights() const { in params_weights()1660 if (!ok()) return ParamsRegions(); in params_weights()1663 ParamsRegions params_biases() const { in params_biases()1664 if (!ok()) return ParamsRegions(); in params_biases()1673 ParamsRegions weights_;1674 ParamsRegions biases_;1738 ParamsRegions ParamsWeightRegions() const override { in ParamsWeightRegions()1739 if (!ok()) return ParamsRegions(); in ParamsWeightRegions()1742 ParamsRegions ParamsBiasRegions() const override { in ParamsBiasRegions()[all …]