Searched defs:DenseUpdate (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | dense_update_functor_gpu.cu.cc | 31 struct DenseUpdate<GPUDevice, T, ASSIGN> { struct
32 void operator()(const GPUDevice& d, typename TTypes<T>::Flat params, in operator ()()
39 struct DenseUpdate<GPUDevice, T, ADD> { struct
40 void operator()(const GPUDevice& d, typename TTypes<T>::Flat params, in operator ()()
47 struct DenseUpdate<GPUDevice, T, SUB> { struct
48 void operator()(const GPUDevice& d, typename TTypes<T>::Flat params, in operator ()()
|
| H A D | dense_update_functor.cc | 34 struct DenseUpdate<CPUDevice, string, ASSIGN> { struct
35 void operator()(const CPUDevice& d, typename TTypes<tstring>::Flat params, in operator ()()
|
| H A D | dense_update_functor.h | 36 struct DenseUpdate { struct
|