Searched refs:compute_elementwise_output_strides (Results 1 – 3 of 3) sorted by relevance
669 strides = utils.compute_elementwise_output_strides(input)695 strides = utils.compute_elementwise_output_strides(input)2090 strides = utils.compute_elementwise_output_strides(a)2697 strides = utils.compute_elementwise_output_strides(a)
514 def compute_elementwise_output_strides(*tensors) -> Tuple[int, ...]: function
5055 stride = utils.compute_elementwise_output_strides(*_maybe_broadcast(*inputs))