Searched refs:cuda_set (Results 1 – 1 of 1) sorted by relevance
20 void cuda_set(T* addr, T value) { in cuda_set() function130 cuda_set<float>(c2_tensor_a.mutable_data<float>() + 1, 123); in TEST()143 cuda_set<float>(c2_tensor.mutable_data<float>(), 123); in TEST()148 cuda_set<float>(c2_tensor.mutable_data<float>() + 1, 234); in TEST()153 cuda_set<float>(c2_tensor.mutable_data<float>() + 2, 345); in TEST()161 cuda_set<float>(c2_tensor.mutable_data<float>() + 3, 456); in TEST()