Home
last modified time | relevance | path

Searched refs:kShape4dDims (Results 1 – 7 of 7) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/
Dhccl_kernel_metadata.cc50 if (op_name == kAllGatherOpName && input_shape.size() <= kShape4dDims) { in GetKernelFormat()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/
Dtbe_kernel_reduce_selecter.cc155 …electer::Is4DShape(const std::vector<size_t> &shape) const { return shape.size() == kShape4dDims; } in Is4DShape()
Dtbe_kernel_broadcast_selecter.cc300 return shape.size() == kShape4dDims; in Is4DShape()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/
Dakg_kernel_attrs_process.cc57 if (origin_shape.size() != kShape4dDims) { in SetAkgAttrsForFive2Four()
/third_party/mindspore/mindspore/lite/tools/converter/parser/
Dparser_utils.cc299 if (weight_value->shape().size() != kShape4dDims) { in UnifyConstConvWeight()
/third_party/mindspore/mindspore/ccsrc/utils/
Dutils.h492 const size_t kShape4dDims = 4; variable
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/
Ddynamic_rnn_grad_fission_v2.cc332 if (origin_input4_shape.size() == kShape4dDims) { in CreateHConcat()