Searched defs:ArgMaxOpModel (Results 1 – 3 of 3) sorted by relevance
61 class ArgMaxOpModel : public ArgBaseOpModel { class63 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::__anon09efa6a00111::ArgMaxOpModel
246 class ArgMaxOpModel : public SingleOpModel, public AcceleratedModel { class248 ArgMaxOpModel(std::initializer_list<int> input_shape, TensorType input_type, in ArgMaxOpModel() function in tflite::__anone115d1810111::ArgMaxOpModel255 ArgMaxOpModel(std::initializer_list<int> input_shape, TensorType input_type, in ArgMaxOpModel() function in tflite::__anone115d1810111::ArgMaxOpModel