Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DEmbeddingLookupTest.cpp73 OperandType OutputOpndTy(Type::TENSOR_FLOAT32, weight_shape); in EmbeddingLookupOpModel() local
74 outputs.push_back(model_.addOperand(&OutputOpndTy)); in EmbeddingLookupOpModel()
DHashtableLookupTest.cpp84 OperandType OutputOpndTy(Type::TENSOR_FLOAT32, out_dim); in HashtableLookupOpModel() local
85 outputs.push_back(model_.addOperand(&OutputOpndTy)); in HashtableLookupOpModel()