Searched defs:self_batch (Results 1 – 2 of 2) sorted by relevance
110 const scalar_t* self_batch = &self_data[b * self_stride]; in apply_triu_tril() local
462 auto self_batch = TensorNames(self_names, 0, num_batch_dims(self_names)); in compute_cdist_outnames() local