Searched defs:constant_axis (Results 1 – 1 of 1) sorted by relevance
34 bool constant_axis, TensorType output_type) in ArgBaseOpModel()85 int axis_value, TensorType axis_type, bool constant_axis, in ArgMaxOpModel()100 int axis_value, TensorType axis_type, bool constant_axis, in ArgMinOpModel()