Searched refs:selected_multi_index (Results 1 – 2 of 2) sorted by relevance
811 std::vector<llvm::Value*> selected_multi_index; in HandleSelectAndScatter() local815 selected_multi_index.push_back(Load(selected_index_address_slot)); in HandleSelectAndScatter()822 selected_multi_index, output_array.GetShape(), source_index.GetType()); in HandleSelectAndScatter()
2393 std::vector<llvm::Value*> selected_multi_index; in EmitSelectAndScatterFromMlir() local2397 selected_multi_index.push_back(Load(selected_index_address_slot)); in EmitSelectAndScatterFromMlir()2403 IrArray::Index selected_index(selected_multi_index, output_shape, in EmitSelectAndScatterFromMlir()