Searched defs:to_dim (Results 1 – 3 of 3) sorted by relevance
386 for (int64_t to_dim = common_factors[i + 1].second - 1; in ConvertDimensionNumbers() local
371 auto to_dim = at::maybe_wrap_dim(std::get<int64_t>(inputs[2]), A.ndim()); in computeTranspose() local
719 int64_t to_dim = broadcast_dimensions[from_dim]; in BinaryOp() local