utils.h File Reference

#include <metal_stdlib>

Go to the source code of this file.

Functions
METAL_FUNC ulong2	elem_to_loc_broadcast (uint elem, constant const int *shape, constant const int64_t *a_strides, constant const int64_t *b_strides, int ndim)
METAL_FUNC ulong3	elem_to_loc_broadcast (uint elem, constant const int *shape, constant const int64_t *a_strides, constant const int64_t *b_strides, constant const int64_t *c_strides, int ndim)

Function Documentation

elem_to_loc_broadcast() [1/2]

METAL_FUNC ulong3 elem_to_loc_broadcast	(	uint	elem,
		constant const int *	shape,
		constant const int64_t *	a_strides,
		constant const int64_t *	b_strides,
		constant const int64_t *	c_strides,
		int	ndim )

elem_to_loc_broadcast() [2/2]

METAL_FUNC ulong2 elem_to_loc_broadcast	(	uint	elem,
		constant const int *	shape,
		constant const int64_t *	a_strides,
		constant const int64_t *	b_strides,
		int	ndim )