Searched refs:RankOneSelect (Results 1 – 6 of 6) sorted by relevance
113 TF_LITE_SWITCH(input_x->type, RankOneSelect); in SelectEval()
1028 inline void RankOneSelect(const D* input_condition_data, in RankOneSelect() function1033 RankOneSelect(DimsToShape(input_condition_dims), input_condition_data, in RankOneSelect()
4314 void RankOneSelect(const RuntimeShape& input_condition_shape, in RankOneSelect() function
65 using reference_ops::RankOneSelect;
82 using reference_ops::RankOneSelect;
76 using reference_ops::RankOneSelect;