Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/auto_parallel/
Dstage_compute.cc199 Shapes param_shapes = GetNodeShape(p); in GetVocabAndHiddenSize() local
232 Shapes param_shapes = GetNodeShape(cnode); in GetSeqLengthAndAttentionHeads() local
238 Shapes param_shapes = GetNodeShape(cnode); in GetSeqLengthAndAttentionHeads() local
295 Shapes param_shapes = GetNodeShape(cnode); in GetPerBatch() local
411 Shapes param_shapes = GetNodeShape(p); in GetExpansionRatio() local