Searched refs:axes_di (Results 1 – 1 of 1) sorted by relevance
142 typename Eigen::array<bool, NDIMS> axes_di; in HandleReverseCase() local144 axes_di[i] = dims(i); in HandleReverseCase()147 input.tensor<T, NDIMS>(), axes_di, in HandleReverseCase()219 typename Eigen::array<bool, NDIMS> axes_di; in HandleReverseV2Case() local221 axes_di[i] = axes[i]; in HandleReverseV2Case()224 input.tensor<T, NDIMS>(), axes_di, in HandleReverseV2Case()