Searched defs:_dim_arange (Results 1 – 3 of 3) sorted by relevance
846 def _dim_arange(g: jit_utils.GraphContext, like, dim): function
4669 def _dim_arange(g: jit_utils.GraphContext, like, dim): function
179 Tensor _dim_arange(const Tensor& like, int64_t dim) { in _dim_arange() function