Searched refs:lookup_shape (Results 1 – 1 of 1) sorted by relevance
61 HashtableLookupOpModel(std::initializer_list<uint32_t> lookup_shape, in HashtableLookupOpModel() argument71 OperandType LookupTy(Type::TENSOR_INT32, lookup_shape); in HashtableLookupOpModel()83 std::vector<uint32_t> out_dim(lookup_shape.begin(), lookup_shape.end()); in HashtableLookupOpModel()89 OperandType HitsOpndTy(Type::TENSOR_QUANT8_ASYMM, lookup_shape, 1.f, 0); in HashtableLookupOpModel()102 Hits_.insert(Hits_.end(), multiAll(lookup_shape), 0); in HashtableLookupOpModel()