Searched refs:kConv2DBackpropInputOpName (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | cudnn_inplace_fusion.cc | 45 std::map<string, uint32_t> kInplaceOpNames = {{kConv2DBackpropInputOpName, 0}, {kBatchNormGradWithA…
53 …{kConv2DBackpropInputOpName, 0}, {kmaxPoolGradOpName, 2}, {kBatchNormGradWithAddAndActivation, 0}};
120 if (AnfAlgo::GetCNodeName(first_node) != kConv2DBackpropInputOpName || in GetCoverIndex()
121 AnfAlgo::GetCNodeName(second_node) != kConv2DBackpropInputOpName) { in GetCoverIndex()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/ |
| D | conv_transpose_to_conv_bp.cc | 48 prim->Named::operator=(Named(kConv2DBackpropInputOpName)); in Process()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_adapter.cc | 66 std::unordered_set<std::string> TbeAdapter::input_order_adjusted_ops_ = {kConv2DBackpropInputOpName,
86 if (op_name == kConv2DBackpropInputOpName) { in DynamicInputAdjusted()
288 return (op_name == kConv2DBackpropInputOpName || op_name == kConv2DOpName); in IsSpecialFusionComputeNode()
352 … {kConv2DBackpropInputOpName, "Conv2d_backprop_input"}, in GetNodeFusionType()
|
| D | tbe_adapter.h | 103 if (op_name == kConv2DBackpropInputOpName) { in DynamicInputAdjusted()
|
| D | tbe_kernel_build.cc | 1221 return (node_name == kConv2DBackpropInputOpName || node_name == kConv2DOpName); in GetInputLayers()
1365 … {kConv2DBackpropInputOpName, "Conv2d_backprop_input"}, in GetNodeFusionType()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/ |
| D | set_fracz_group_attr.cc | 218 if (node_name == kConv2DOpName || node_name == kConv2DBackpropInputOpName || in Run()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/trt_pass/ |
| D | graph_partitioner.cc | 38 {kConv2DOpName, {1}}, {kBatchNorm, {1, 2, 3, 4}}, {kConv2DBackpropInputOpName, {1}}}; in WeightCheck()
|
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | utils.h | 187 constexpr auto kConv2DBackpropInputOpName = "Conv2DBackpropInput"; variable
|