Searched defs:ArgMaxOpModel (Results 1 – 3 of 3) sorted by relevance
59 class ArgMaxOpModel : public ArgBaseOpModel { class61 ArgMaxOpModel(std::initializer_list<int> input_shape, TensorType input_type) in ArgMaxOpModel() function in tflite::ArgMaxOpModel
82 class ArgMaxOpModel : public ArgBaseOpModel { class84 ArgMaxOpModel(std::initializer_list<int> input_shape, TensorType input_type, in ArgMaxOpModel() function in tflite::__anonfd9765560111::ArgMaxOpModel
242 class ArgMaxOpModel : public SingleOpModel, public AcceleratedModel { class244 ArgMaxOpModel(std::initializer_list<int> input_shape, TensorType input_type, in ArgMaxOpModel() function in tflite::__anone5dffd370111::ArgMaxOpModel251 ArgMaxOpModel(std::initializer_list<int> input_shape, TensorType input_type, in ArgMaxOpModel() function in tflite::__anone5dffd370111::ArgMaxOpModel