Home
last modified time | relevance | path

Searched refs:lookup_shape (Results 1 – 1 of 1) sorted by relevance

/frameworks/ml/nn/common/operations/
DHashtableLookupTest.cpp59 HashtableLookupOpModel(std::initializer_list<uint32_t> lookup_shape, in HashtableLookupOpModel() argument
69 OperandType LookupTy(Type::TENSOR_INT32, lookup_shape); in HashtableLookupOpModel()
81 std::vector<uint32_t> out_dim(lookup_shape.begin(), lookup_shape.end()); in HashtableLookupOpModel()
87 OperandType HitsOpndTy(Type::TENSOR_QUANT8_ASYMM, lookup_shape, 1.f, 0); in HashtableLookupOpModel()
100 Hits_.insert(Hits_.end(), multiAll(lookup_shape), 0); in HashtableLookupOpModel()