Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/
Dmaxpool_with_argmax_unify_mindir.cc32 constexpr size_t kAlignBytes = 16; variable
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()
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dmemory_manager.cc28 constexpr size_t kAlignBytes = 32; variable
31 return (input_size + kMemAlignSize + kAlignBytes - 1) / kMemAlignSize * kMemAlignSize; in GetCommonAlignSize()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dkernel_build_ascend.cc136 constexpr size_t kAlignBytes = 32 - 1; in CalCleanZerosSize() local
142 …auto clean_item = (output_men_size.at(index) + kMemAlignSize + kAlignBytes) / kMemAlignSize * kMem… in CalCleanZerosSize()
151 …auto clean_item = (workspace_men_sizes.at(index) + kMemAlignSize + kAlignBytes) / kMemAlignSize * … in CalCleanZerosSize()