Home
last modified time | relevance | path

Searched refs:kMaxShapeDims (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/
Daicpu_util.h81 const uint32_t kMaxShapeDims = 8; variable
84 int64_t dims[kMaxShapeDims];
Daicpu_kernel_build.cc326 if (input_shape.size() < kMaxShapeDims) { in SetExtInfoInputShapeType()
353 if (output_shape_index < kMaxShapeDims) { in SetExtInfoOutputShapeType()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/
Daicpu_ext_info_handle.cc215 if (shape.empty() || shape.size() > kernel::kMaxShapeDims) { in UpdateShapeAndType()
224 if (index < kernel::kMaxShapeDims) { in UpdateShapeAndType()