Home
last modified time | relevance | path

Searched refs:kShape5dDims (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/
Dtbe_kernel_reduce_selecter.cc157 …electer::Is5DShape(const std::vector<size_t> &shape) const { return shape.size() == kShape5dDims; } in Is5DShape()
Dtbe_kernel_broadcast_selecter.cc304 return shape.size() == kShape5dDims; in Is5DShape()
Dtbe_kernel_select.cc261 if (k3DFormatSet.find(format) != k3DFormatSet.end() && shape.size() > kShape5dDims) { in IsShapeMatchFormat()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dtbe_adapter.cc319 …sion_data_type == kFusionAddN || fusion_data_type == kFusionAdd) && shape.size() == kShape5dDims) { in FusionDescJsonPass()
/third_party/mindspore/mindspore/ccsrc/utils/
Dutils.h495 const size_t kShape5dDims = 5; variable