Searched refs:WeightTy (Results 1 – 2 of 2) sorted by relevance
54 OperandType WeightTy(Type::TENSOR_FLOAT32, weight_shape); in LSHProjectionOpModel() local55 inputs.push_back(model_.addOperand(&WeightTy)); in LSHProjectionOpModel()
139 OperandType WeightTy(Type::TENSOR_FLOAT32, {units_, input_size_}); in BasicRNNOpModel() local140 inputs.push_back(model_.addOperand(&WeightTy)); in BasicRNNOpModel()