Searched refs:IndexComponent (Results 1 – 1 of 1) sorted by relevance
201 enum class IndexComponent { Ungathered, GatheredFirst, GatheredSecond }; in FoldGatherOfGather() enum203 std::vector<IndexComponent> simulated_index(a->shape().dimensions_size(), in FoldGatherOfGather()204 IndexComponent::Ungathered); in FoldGatherOfGather()210 IndexComponent::GatheredFirst); in FoldGatherOfGather()217 IndexComponent::GatheredSecond); in FoldGatherOfGather()226 for (IndexComponent simulation_dim : simulated_index) { in FoldGatherOfGather()227 if (simulation_dim == IndexComponent::GatheredSecond) { in FoldGatherOfGather()230 if (simulation_dim != IndexComponent::Ungathered) { in FoldGatherOfGather()238 if (simulated_index[i] != IndexComponent::Ungathered) { in FoldGatherOfGather()890 enum class IndexComponent { Broadcasted, NotBroadcasted }; in ComputeArrayForElementwiseBinaryOp() enum[all …]