Searched defs:GetBatchAxisForInput (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pipeline_transformer/ | ||
| D | pipeline_interleave.cc | 212 size_t PipelineInterleave::GetBatchAxisForInput(const AnfNodeIndexSet &input_node_users) const { in GetBatchAxisForInput() function in mindspore::parallel::PipelineInterleave |
| D | pipeline_transformer.cc | 361 size_t PipelineTransformer::GetBatchAxisForInput(const AnfNodeIndexSet &input_node_users) const { in GetBatchAxisForInput() function in mindspore::parallel::PipelineTransformer |