Searched refs:weight_shape (Results 1 – 2 of 2) sorted by relevance
57 std::initializer_list<uint32_t> weight_shape) { in EmbeddingLookupOpModel() argument58 auto it = weight_shape.begin(); in EmbeddingLookupOpModel()68 OperandType ValueTy(Type::TENSOR_FLOAT32, weight_shape); in EmbeddingLookupOpModel()73 OperandType OutputOpndTy(Type::TENSOR_FLOAT32, weight_shape); in EmbeddingLookupOpModel()82 Value_.insert(Value_.end(), multiAll(weight_shape), 0.f); in EmbeddingLookupOpModel()83 Output_.insert(Output_.end(), multiAll(weight_shape), 0.f); in EmbeddingLookupOpModel()
45 std::initializer_list<uint32_t> weight_shape) in LSHProjectionOpModel() argument53 OperandType WeightTy(Type::TENSOR_FLOAT32, weight_shape); in LSHProjectionOpModel()