Home
last modified time | relevance | path

Searched refs:format_attr (Results 1 – 13 of 13) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dbatch_norm_relu_fusion.cc51 auto format_attr = AnfAlgo::GetCNodePrimitive(batch_norm)->GetAttr("format"); in Process() local
52 MS_EXCEPTION_IF_NULL(format_attr); in Process()
53 auto format = GetValue<std::string>(format_attr); in Process()
Dbatch_norm_add_relu_fusion.cc54 auto format_attr = AnfAlgo::GetCNodePrimitive(batch_norm)->GetAttr("format"); in Process() local
55 MS_EXCEPTION_IF_NULL(format_attr); in Process()
56 auto format = GetValue<std::string>(format_attr); in Process()
Dpost_batch_norm_add_relu_fusion.cc54 auto format_attr = AnfAlgo::GetCNodePrimitive(batch_norm)->GetAttr("format"); in Process() local
55 MS_EXCEPTION_IF_NULL(format_attr); in Process()
56 auto format = GetValue<std::string>(format_attr); in Process()
Dbatch_norm_relu_grad_fusion.cc47 auto format_attr = AnfAlgo::GetCNodePrimitive(node)->GetAttr("format"); in Process() local
48 MS_EXCEPTION_IF_NULL(format_attr); in Process()
49 auto format = GetValue<std::string>(format_attr); in Process()
Dbatch_norm_add_relu_grad_fusion.cc118 auto format_attr = AnfAlgo::GetCNodePrimitive(node)->GetAttr("format"); in PatternCheck() local
119 MS_EXCEPTION_IF_NULL(format_attr); in PatternCheck()
120 auto format = GetValue<std::string>(format_attr); in PatternCheck()
/third_party/python/Lib/test/
Dpythoninfo.py187 def format_attr(attr, value): function
201 copy_attributes(info_add, os, 'os.%s', attributes, formatter=format_attr)
348 def format_attr(attr, value): function
360 formatter=format_attr)
492 def format_attr(attr, value): function
505 copy_attributes(info_add, ssl, 'ssl.%s', attributes, formatter=format_attr)
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/
Dpooling_gpu_kernel.h86 auto format_attr = GetAttr<std::string>(kernel_node, "format"); in Init() local
87 if (Anyone(format_attr, kOpFormat_NHWC, kOpFormat_NDHWC)) { in Init()
88 data_format_ = format_attr; in Init()
Dconv2d_grad_input_gpu_kernel.h124 auto format_attr = GetAttr<std::string>(kernel_node, "format"); in Init() local
125 if (format_attr == kOpFormat_NHWC) { in Init()
138 if (format_attr == kOpFormat_NCHW) { in Init()
Dbatch_norm_gpu_kernel.h137 auto format_attr = GetAttr<std::string>(kernel_node, "format"); in Init() local
138 if (format_attr == kOpFormat_NHWC) { in Init()
Dbatch_norm_grad_gpu_kernel.h164 auto format_attr = GetAttr<std::string>(kernel_node, "format"); in Init() local
165 if (format_attr == kOpFormat_NHWC) { in Init()
Dconv3d_transpose_gpu_kernel.h119 auto format_attr = GetAttr<std::string>(kernel_node, "format"); in Init() local
120 if (format_attr == kOpFormat_NDHWC) { in Init()
132 if (format_attr == kOpFormat_NCDHW) { in Init()
Dconv2d_gpu_kernel.h87 auto format_attr = GetAttr<std::string>(kernel_node, "format"); in Init() local
88 if (format_attr == kOpFormat_NHWC) { in Init()
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Dinfershape_pass.cc53 auto format_attr = primitive->GetAttr(ops::kFormat); in GetCNodeCertainInputFormat() local
54 MS_CHECK_TRUE_MSG(format_attr != nullptr, lite::RET_NULL_PTR, "GetAttr Failed"); in GetCNodeCertainInputFormat()
55 *format = static_cast<mindspore::Format>(GetValue<int64_t>(format_attr)); in GetCNodeCertainInputFormat()