Home
last modified time | relevance | path

Searched refs:ParamsRegions (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc966 typedef dnn::RnnDescriptor::ParamsRegions ParamsRegions; typedef in stream_executor::gpu::__anonf41f84630311::CudnnRnnParamsDescriptor
969 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()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h171 typedef std::vector<ParamsRegion> ParamsRegions;
174 virtual ParamsRegions ParamsWeightRegions() const { return ParamsRegions(); }
175 virtual ParamsRegions ParamsBiasRegions() const { return ParamsRegions(); }
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc1647 typedef dnn::RnnDescriptor::ParamsRegions ParamsRegions; typedef in stream_executor::gpu::MIOpenRnnParamsDescriptor
1659 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 …]