Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/tests/st/ops/
Dtest_ops_gather_ext.py159 def _foreach_run(x, dim, index, batch_axis): argument
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/_vmap/
Dvmap_base.py483 def _get_reduce_out_dim(x_dim, batch_axis, keep_dims=False): argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pipeline_transformer/
Dpipeline_interleave.cc227 size_t batch_axis = 0; in GetBatchAxisForInput() local
265 auto batch_axis = GetBatchAxisForInput(data_users); in LabelMicroBatch() local
Dpipeline_transformer.cc376 size_t batch_axis = 0; in GetBatchAxisForInput() local
434 auto batch_axis = GetBatchAxisForInput(data_users); in LabelMicroBatch() local