Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/dtensor/mlir/expansions/
Ddataparallel_spmd_expander.cc104 int batch_rank = in MergeBatchLayouts() local
158 int batch_rank = operand_layouts[first_batcharg_index].rank() - in IntermediateBatchLayout() local