Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/kernel/
Dreduce.c264 int tmp_input_shape[MAX_SHAPE_SIZE]; in ReduceCalculateInnerOuterSize() local
292 int tmp_input_shape[MAX_SHAPE_SIZE]; in ReduceCalculateTmpBufferSize() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/
Dtile_info.cc272 Shape tmp_input_shape = full_multiples_; in GenerateOpStrategies() local
Doperator_info.cc2092 Shape tmp_input_shape; in GenerateStrategiesForDependentInputs() local