Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Ddataparallel_spmd_expander.h50 explicit DataparallelSPMDExpander(llvm::DenseMap<int, int> batchable_operands, in DataparallelSPMDExpander() argument
H A Ddataparallel_spmd_expander.cc150 const llvm::DenseMap<int, int>& batchable_operands, in IntermediateBatchLayout() argument