Searched refs:simulated_index (Results 1 – 1 of 1) sorted by relevance
203 std::vector<IndexComponent> simulated_index(a->shape().dimensions_size(), in FoldGatherOfGather() local207 EraseAt(&simulated_index, source->source_dim()); in FoldGatherOfGather()209 simulated_index.insert(simulated_index.begin() + gather_dim, in FoldGatherOfGather()214 EraseAt(&simulated_index, source_dim); in FoldGatherOfGather()216 simulated_index.insert(simulated_index.begin() + output_dim, in FoldGatherOfGather()226 for (IndexComponent simulation_dim : simulated_index) { in FoldGatherOfGather()237 for (int64 i = 0, e = simulated_index.size(); i < e; i++) { in FoldGatherOfGather()238 if (simulated_index[i] != IndexComponent::Ungathered) { in FoldGatherOfGather()891 std::vector<IndexComponent> simulated_index( in ComputeArrayForElementwiseBinaryOp() local894 simulated_index[broadcast_dim] = IndexComponent::NotBroadcasted; in ComputeArrayForElementwiseBinaryOp()[all …]