Searched defs:logitsType (Results 1 – 2 of 2) sorted by relevance
84 const llm_helper::LLMType logitsType, in argmax()
47 OperandType logitsType(Type::TENSOR_FLOAT32, {batch_size_, class_size_}); in MultinomialOpModel() local