Searched defs:rank_diff (Results 1 – 2 of 2) sorted by relevance
166 def _run_trainer(rref_t1, t2, ps, rank_diff, sparse): argument180 def _run_trainer_torchscript(rref_t1, t2, ps, rank_diff, sparse): argument
738 int rank_diff = shape_x.size() - shape_y.size(); in AreBroadcastCompatible() local1630 auto rank_diff = ranked_output_type.getRank() - input_type.getRank(); in matchAndRewrite() local