Searched refs:logitsType (Results 1 – 1 of 1) sorted by relevance
42 OperandType logitsType(Type::TENSOR_FLOAT32, {batch_size_, class_size_}); in MultinomialOpModel() local43 inputs.push_back(model_.addOperand(&logitsType)); in MultinomialOpModel()