Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DLSHProjectionTest.cpp79 OperandType OutputTy(Type::TENSOR_INT32, {outShapeDimension}); in LSHProjectionOpModel() local
80 outputs.push_back(model_.addOperand(&OutputTy)); in LSHProjectionOpModel()
DRNNTest.cpp175 OperandType OutputTy(Type::TENSOR_FLOAT32, {batches_, units_}); in BasicRNNOpModel() local
176 outputs.push_back(model_.addOperand(&OutputTy)); in BasicRNNOpModel()