Searched defs:axis_split_step (Results 1 – 2 of 2) sorted by relevance
97 const size_t axis_split_step = output_shape[axis]; in validate() local166 const size_t axis_split_step = output_shape[axis]; in configure() local
119 const size_t axis_split_step = shape[axis] / splits; in compute_reference() local231 const size_t axis_split_step = split_shape[axis]; in compute_reference() local