Searched refs:kDim4 (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | avgpool_3d_grad_fusion.cc | 86 auto fw = fp_shape[kDim4]; in IsVectorImpl()
92 bool flag3 = kw >= fw + pad_list[kDim4] + pad_list[kDim5]; in IsVectorImpl()
135 auto pad_w = pad_list[kDim4] + pad_list[kDim5]; in ConstructMultiplier()
138 auto len_w = ori_input_shape[kDim4] + pad_w; in ConstructMultiplier()
146 for (int64_t wi = 0; wi < grad_shape[kDim4]; wi++) { in ConstructMultiplier()
159 … valid_w = std::min(start_w + kernel_size[kDim2], pad_list[kDim4] + ori_input_shape[kDim4]) - in ConstructMultiplier()
160 std::max(pad_list[kDim4], start_w); in ConstructMultiplier()
|
| D | avgpool_3d_fusion.cc | 71 *kw = kernel_size[kDim4]; in GetKernelSize()
99 *sw = kernel_size[kDim4]; in GetStrideSize()
171 auto pad_w = pad_list[kDim4] + pad_list[kDim5]; in ConstructMultiplier()
188 … auto vaild_w = GetInterSection(start_w, start_w + kw, pad_list[kDim4], pad_list[kDim4] + fw); in ConstructMultiplier()
268 auto fw = SizeToLong(dims_in[kDim4]); in Process()
271 auto dw = SizeToLong(dims_out[kDim4]); in Process()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/ |
| D | bn_grad_unify_mindir.cc | 42 bn_grad_node_inputs[kDim4], in CreateNewBatchNormGrad()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | max_pool3d_grad_grad_fission.cc | 45 int64_t w = ksize[kDim4]; in CreateTensor()
|
| D | cdist_fission.cc | 146 auto broadcast_out = AddBroadCastToNode(graph, cdist_grad_inputs[kDim4], 0, broadcast_to_shape); in Process()
|
| /third_party/mindspore/mindspore/lite/tools/converter/quantizer/ |
| D | quantize_util.cc | 49 constexpr int kDim4 = 4; variable
234 if (weight_shape.size() == kDim4) { // assume Convolution in CanTensorQuantized()
|
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | utils.h | 564 kDim4, enumerator
|
| /third_party/mindspore/mindspore/ccsrc/common/ |
| D | trans.cc | 1431 if (hw_shape.size() < kDim3 || args.device_shape.size() < kDim4) { in NchwToFracNz()
1493 if (hw_shape.size() < kDim3 || args.device_shape.size() < kDim4) { in FracNzToNchw()
|