Searched refs:Set3DWeightMatrix (Results 1 – 1 of 1) sorted by relevance
129 void Set3DWeightMatrix(const std::function<float(int, int, int)>& function) { in Set3DWeightMatrix() function in android::nn::wrapper::EmbeddingLookupOpModel160 m.Set3DWeightMatrix([](int i, int j, int k) { return i + j / 10.0f + k / 100.0f; }); in TEST()