Searched refs:ValueTy (Results 1 – 2 of 2) sorted by relevance
69 OperandType ValueTy(Type::TENSOR_FLOAT32, weight_shape); in EmbeddingLookupOpModel() local70 inputs.push_back(model_.addOperand(&ValueTy)); in EmbeddingLookupOpModel()
76 OperandType ValueTy(Type::TENSOR_FLOAT32, value_shape); in HashtableLookupOpModel() local77 inputs.push_back(model_.addOperand(&ValueTy)); in HashtableLookupOpModel()