Home
last modified time | relevance | path

Searched defs:input_tensor_size (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_class/
Dunique_consecutive_helper.h68 size_t input_tensor_size = input_size_list_[0]; in CalMemSize() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/
Drec_parse_graph.cc94 size_t input_tensor_size = ops[iter_ops]->inputs_shape().size(); in CompleteOperatorInputs() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/optimizer/
Dhelper.cc166 void CheckCNodeInputSize(const CNodePtr &cnode, size_t input_tensor_size) { in CheckCNodeInputSize()