Searched refs:value_init_value (Results 1 – 1 of 1) sorted by relevance
148 XlaOp value_init_value; in ArgMinMax() local150 value_init_value = MaxValue(builder, input_shape.element_type()); in ArgMinMax()152 value_init_value = MinValue(builder, input_shape.element_type()); in ArgMinMax()165 {value_init_value, index_init_value}, reducer, in ArgMinMax()