Searched defs:selected_index (Results 1 – 8 of 8) sorted by relevance
41 for (const int selected_index : result) { in DiversifyByDistance() local
542 int selected_index = selected[i]; in NonMaxSuppressionMultiClassRegularHelper() local672 int selected_index = selected[i]; in NonMaxSuppressionMultiClassFastHelper() local
570 for (const auto& selected_index : selected) { in NonMaxSuppressionMultiClassRegularHelper() local703 for (const auto& selected_index : selected) { in NonMaxSuppressionMultiClassFastHelper() local
449 …unsigned selected_index = get_default_device(info, selection, physical_device_count, physical_devi… in device_select_EnumeratePhysicalDevices() local
254 int64 selected_index = 0; in Compute() local
821 llvm_ir::IrArray::Index selected_index( in HandleSelectAndScatter() local
2403 IrArray::Index selected_index(selected_multi_index, output_shape, in EmitSelectAndScatterFromMlir() local
951 Value selected_index = builder->create<SelectOp>( in BuildArgMinMaxReductionBody() local