Searched refs:input_rest_by_depth (Results 1 – 2 of 2) sorted by relevance
86 auto input_rest_by_depth = input.reshape(rest_by_depth).template cast<U>(); in operator ()() local96 (input_rest_by_depth - in operator ()()
586 auto input_rest_by_depth = input.reshape(rest_by_depth).template cast<U>(); in operator ()() local597 (input_rest_by_depth - in operator ()()