Home
last modified time | relevance | path

Searched defs:set_data_format (Results 1 – 13 of 13) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/fusion/
Davg_pool_fusion.cc52 void AvgPoolFusion::set_data_format(const int64_t &data_format) { in set_data_format() function in mindspore::ops::AvgPoolFusion
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/auto_generate/
Dgen_lite_ops.cc195 void AvgPoolGrad::set_data_format(const int64_t &data_format) { (void)this->AddAttr("da… in set_data_format() function in mindspore::ops::AvgPoolGrad
214 void AvgPool::set_data_format(const int64_t &data_format) { (void)this->AddAttr("data_f… in set_data_format() function in mindspore::ops::AvgPool
254 void BatchNormGradGrad::set_data_format(const int64_t &data_format) { (void)this->AddAt… in set_data_format() function in mindspore::ops::BatchNormGradGrad
269 void BatchNormGrad::set_data_format(const int64_t &data_format) { (void)this->AddAttr("… in set_data_format() function in mindspore::ops::BatchNormGrad
284 void BatchNormGradWithActivation::set_data_format(const int64_t &data_format) { (void)t… in set_data_format() function in mindspore::ops::BatchNormGradWithActivation
303 void BatchNorm::set_data_format(const int64_t &data_format) { (void)this->AddAttr("data… in set_data_format() function in mindspore::ops::BatchNorm
322 void BatchNormWithActivation::set_data_format(const int64_t &data_format) { (void)this-… in set_data_format() function in mindspore::ops::BatchNormWithActivation
341 void BatchNormWithAddAndActivation::set_data_format(const int64_t &data_format) { (void… in set_data_format() function in mindspore::ops::BatchNormWithAddAndActivation
351 void BiasAddGrad::set_data_format(const int64_t &data_format) { (void)this->AddAttr("da… in set_data_format() function in mindspore::ops::BiasAddGrad
358 void BiasAdd::set_data_format(const int64_t &data_format) { (void)this->AddAttr("data_f… in set_data_format() function in mindspore::ops::BiasAdd
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/grad/
Dfractional_max_pool_grad_with_fixed_ksize.cc153 void FractionalMaxPoolGradWithFixedKsize::set_data_format(const std::string data_format) { in set_data_format() function in mindspore::ops::FractionalMaxPoolGradWithFixedKsize
Dfractional_max_pool3d_grad_with_fixed_ksize.cc192 void FractionalMaxPool3DGradWithFixedKsize::set_data_format(const std::string data_format) { in set_data_format() function in mindspore::ops::FractionalMaxPool3DGradWithFixedKsize
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dfractional_max_pool3d_with_fixed_ksize.cc246 void FractionalMaxPool3DWithFixedKsize::set_data_format(const std::string data_format) { in set_data_format() function in mindspore::ops::FractionalMaxPool3DWithFixedKsize
Dfractional_max_pool_with_fixed_ksize.cc209 void FractionalMaxPoolWithFixedKsize::set_data_format(const std::string data_format) { in set_data_format() function in mindspore::ops::FractionalMaxPoolWithFixedKsize
Ddeformable_conv2d.cc229 void DeformableConv2d::set_data_format(const std::string &data_format) { in set_data_format() function in mindspore::ops::DeformableConv2d
Ddeformable_offsets.cc239 void DeformableOffsets::set_data_format(const std::string &data_format) { in set_data_format() function in mindspore::ops::DeformableOffsets
Dconv3d_transpose.cc134 void Conv3DTranspose::set_data_format(const Format &format) { in set_data_format() function in mindspore::ops::Conv3DTranspose
Dconv3d.cc123 void Conv3D::set_data_format(const Format &format) { in set_data_format() function in mindspore::ops::Conv3D
/third_party/backends/backend/
Depson.h91 unsigned char set_data_format; member
Depson2.h199 unsigned char set_data_format; member
Depson.c1123 #define set_data_format(s,v) set_cmd( s,(s)->hw->cmd->set_data_format, v) macro