Searched refs:kDim3 (Results 1 – 10 of 10) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/ |
| D | avg_pool_grad_unify_mindir.cc | 82 x_shape[kDim3] + pad_left + pad_right}; in GetAssistInputMatrix()
114 …windowed_output_size(x_shape[kDim3], k_size[kDim3], stride[kDim3], pad_mode, &pad_left, &pad_right… in CreateMeanMatrixValueNode()
123 for (int64_t j = w * stride[kDim3]; j < w * stride[kDim3] + k_size[kDim3]; ++j) { in CreateMeanMatrixValueNode()
139 auto dst_size = LongToSize(output_shape[kDim2]) * LongToSize(output_shape[kDim3]) * kFloat32Len; in CreateMeanMatrixValueNode()
165 std::vector<int64_t> kernel_shape = {1, x_shape[kDim1], k_size[kDim2], k_size[kDim3]}; in CreateKernelMatrixValueNode()
|
| D | maxpool_with_argmax_unify_mindir.cc | 72 …argmax_shape[kDim3] = (output_shape[kDim2] * output_shape[kDim3] + kAlignBytes - 1) / kAlignBytes … in Process()
105 …argmax_shape[kDim3] = (argmax_shape[kDim2] * argmax_shape[kDim3] + kAlignBytes - 1) / kAlignBytes … in Process()
|
| D | bn_grad_unify_mindir.cc | 41 bn_grad_node_inputs[kDim3], in CreateNewBatchNormGrad()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | avgpool_3d_grad_fusion.cc | 85 auto fh = fp_shape[kDim3]; in IsVectorImpl()
91 bool flag2 = kh >= fh + pad_list[kDim2] + pad_list[kDim3]; in IsVectorImpl()
134 auto pad_h = pad_list[kDim2] + pad_list[kDim3]; in ConstructMultiplier()
137 auto len_h = ori_input_shape[kDim3] + pad_h; in ConstructMultiplier()
144 for (int64_t hi = 0; hi < grad_shape[kDim3]; hi++) { in ConstructMultiplier()
157 … valid_h = std::min(start_h + kernel_size[kDim1], pad_list[kDim2] + ori_input_shape[kDim3]) - in ConstructMultiplier()
|
| D | avgpool_3d_fusion.cc | 70 *kh = kernel_size[kDim3]; in GetKernelSize()
98 *sh = kernel_size[kDim3]; in GetStrideSize()
170 auto pad_h = pad_list[kDim2] + pad_list[kDim3]; in ConstructMultiplier()
267 auto fh = SizeToLong(dims_in[kDim3]); in Process()
270 auto dh = SizeToLong(dims_out[kDim3]); in Process()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | cdist_fission.cc | 141 auto y_shape = AnfAlgo::GetOutputInferShape(cdist_grad_inputs[kDim3], 0); in Process()
145 …auto broadcast_input_y = AddBroadCastToNode(graph, cdist_grad_inputs[kDim3], kInputYDimR, broadcas… in Process()
|
| D | max_pool3d_grad_grad_fission.cc | 44 int64_t h = ksize[kDim3]; in CreateTensor()
|
| D | space_to_depth_split.cc | 50 int64_t window_size = assist_input_shape[kDim2] * assist_input_shape[kDim3]; in CreateTensor()
|
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | utils.h | 563 kDim3, 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()
|