Searched refs:ParamsRegions (Results 1 – 3 of 3) sorted by relevance
1025 typedef dnn::RnnDescriptor::ParamsRegions ParamsRegions; typedef in stream_executor::gpu::__anon423856410311::CudnnRnnParamsDescriptor1028 ParamsRegions weights, ParamsRegions biases) in CudnnRnnParamsDescriptor()1044 ParamsRegions params_weights() const { return weights_; } in params_weights()1045 ParamsRegions params_biases() const { return biases_; } in params_biases()1050 ParamsRegions weights_;1051 ParamsRegions biases_;1221 ParamsRegions ParamsWeightRegions() const override { in ParamsWeightRegions()1224 ParamsRegions ParamsBiasRegions() const override { in ParamsBiasRegions()1288 dnn::RnnDescriptor::ParamsRegions* weights) { in CheckAndFetchProjectionWeights()1405 ParamsRegions weights; in Create()[all …]
163 typedef std::vector<ParamsRegion> ParamsRegions; typedef166 virtual ParamsRegions ParamsWeightRegions() const { return ParamsRegions(); } in ParamsWeightRegions()167 virtual ParamsRegions ParamsBiasRegions() const { return ParamsRegions(); } in ParamsBiasRegions()
1728 typedef dnn::RnnDescriptor::ParamsRegions ParamsRegions; typedef in stream_executor::gpu::MIOpenRnnParamsDescriptor1740 ParamsRegions params_weights() const { in params_weights()1741 if (!ok()) return ParamsRegions(); in params_weights()1744 ParamsRegions params_biases() const { in params_biases()1745 if (!ok()) return ParamsRegions(); in params_biases()1754 ParamsRegions weights_;1755 ParamsRegions biases_;1819 ParamsRegions ParamsWeightRegions() const override { in ParamsWeightRegions()1820 if (!ok()) return ParamsRegions(); in ParamsWeightRegions()1823 ParamsRegions ParamsBiasRegions() const override { in ParamsBiasRegions()[all …]