Home
last modified time | relevance | path

Searched refs:mergeMultiContext (Results 1 – 2 of 2) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/scala-package/core/src/main/scala/org/apache/mxnet/module/
H A DDataParallelExecutorGroup.scala85 private def mergeMultiContext(outputs: IndexedSeq[IndexedSeq[NDArray]], majorAxis: Seq[Int]) method
536 DataParallelExecutorGroup.mergeMultiContext(getOutputs(), outputLayouts)
559 DataParallelExecutorGroup.mergeMultiContext(getInputGrads(), dataLayouts)
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/scala-package/core/src/main/scala/org/apache/mxnet/module/
H A DDataParallelExecutorGroup.scala85 private def mergeMultiContext(outputs: IndexedSeq[IndexedSeq[NDArray]], majorAxis: Seq[Int]) method
536 DataParallelExecutorGroup.mergeMultiContext(getOutputs(), outputLayouts)
559 DataParallelExecutorGroup.mergeMultiContext(getInputGrads(), dataLayouts)