Searched refs:out_dim (Results 1 – 1 of 1) sorted by relevance
83 std::vector<uint32_t> out_dim(lookup_shape.begin(), lookup_shape.end()); in HashtableLookupOpModel() local84 out_dim.push_back(features_); in HashtableLookupOpModel()86 OperandType OutputOpndTy(Type::TENSOR_FLOAT32, out_dim); in HashtableLookupOpModel()101 Output_.insert(Output_.end(), multiAll(out_dim), 0.f); in HashtableLookupOpModel()