Searched defs:input_logical_rank (Results 1 – 2 of 2) sorted by relevance
109 const auto input_logical_rank = rankWithoutBatchDim(input, input_bdim); in batch_norm_batch_rule() local510 const auto input_logical_rank = rankWithoutBatchDim(input, input_bdim); in native_layer_norm_batch_rule() local
457 auto input_logical_rank = rankWithoutBatchDim(input, input_bdim); in atol_rtol_tensor_batch_rule() local