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