Searched refs:columns_ (Results 1 – 1 of 1) sorted by relevance
61 columns_ = *it++; in EmbeddingLookupOpModel()131 for (uint32_t j = 0; j < columns_; j++) { in Set3DWeightMatrix()133 Value_[(i * columns_ + j) * features_ + k] = function(i, j, k); in Set3DWeightMatrix()144 uint32_t columns_; member in android::nn::wrapper::EmbeddingLookupOpModel