Searched defs:tensor_logical_rank (Results 1 – 4 of 4) sorted by relevance
105 auto tensor_logical_rank = rankWithoutBatchDim(tensor, tensor_batch_dim); in binary_pointwise_inplace_batch_rule() local127 auto tensor_logical_rank = rankWithoutBatchDim(tensor, tensor_batch_dim); in comparison_pointwise_batch_rule() local149 auto tensor_logical_rank = rankWithoutBatchDim(self, self_bdim); in where_self_batch_rule() local
75 auto tensor_logical_rank = rankWithoutBatchDim(tensor, has_bdim); in maybePadToLogicalRank() local176 auto tensor_logical_rank = rankWithoutBatchDim(tensor, tensor_batch_dim); in _binary_pointwise_helper() local
33 auto tensor_logical_rank = rankWithoutBatchDim(tensor, has_bdim); in padRight() local
1024 auto tensor_logical_rank = rankWithoutBatchDim(self, self_bdim); in binary_pointwise_align() local