Searched refs:input_condition_shape (Results 1 – 1 of 1) sorted by relevance
4300 void Select(const RuntimeShape& input_condition_shape, in Select() argument4306 input_condition_shape, input_x_shape, input_y_shape, output_shape); in Select()4314 void RankOneSelect(const RuntimeShape& input_condition_shape, in RankOneSelect() argument4319 const int64_t outer_size = input_condition_shape.FlatSize(); in RankOneSelect()4335 void SelectTrueCoords(const RuntimeShape& input_condition_shape, in SelectTrueCoords() argument4337 const size_t size = input_condition_shape.FlatSize(); in SelectTrueCoords()4338 const size_t cond_rank = input_condition_shape.DimensionsCount(); in SelectTrueCoords()4343 dims_to_count[i] = cur_flat_size / input_condition_shape.Dims(i); in SelectTrueCoords()