Searched refs:DenseUpdate (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | dense_update_functor_gpu.cu.cc | 31 struct DenseUpdate<GPUDevice, T, ASSIGN> { struct
39 struct DenseUpdate<GPUDevice, T, ADD> { struct
47 struct DenseUpdate<GPUDevice, T, SUB> { struct
57 template struct functor::DenseUpdate<GPUDevice, T, ADD>; \
58 template struct functor::DenseUpdate<GPUDevice, T, SUB>;
66 template struct functor::DenseUpdate<GPUDevice, T, ASSIGN>;
|
| D | dense_update_functor.h | 36 struct DenseUpdate { struct
42 struct DenseUpdate<CPUDevice, T, ADD> {
50 struct DenseUpdate<CPUDevice, T, SUB> {
58 struct DenseUpdate<CPUDevice, T, ASSIGN> {
|
| D | dense_update_functor.cc | 34 struct DenseUpdate<CPUDevice, string, ASSIGN> { struct
76 functor::DenseUpdate<CPUDevice, T, ASSIGN> copy_functor_; \
111 functor::DenseUpdate<GPUDevice, T, ASSIGN> copy_functor_; \
|
| D | dense_update_ops.cc | 39 functor::DenseUpdate<Device, T, ASSIGN> copy; in Copy()
80 functor::DenseUpdate<Device, T, OP> update_functor; in DoUpdate()
|
| D | training_op_helpers.h | 63 functor::DenseUpdate<Device, T, ASSIGN> copy_functor; in EnsureSparseVariableAccess()
222 functor::DenseUpdate<Device, T, ASSIGN> copy_functor; in PrepareToUpdateVariable()
|
| D | resource_variable_ops.cc | 269 void DenseUpdate<GPUDevice, type, ASSIGN>::operator()( \
272 extern template struct DenseUpdate<GPUDevice, type, ASSIGN>; \
402 functor::DenseUpdate<Device, T, ASSIGN> copy_functor; in Compute()
544 functor::DenseUpdate<Device, T, Op> update_functor; in Compute()
677 functor::DenseUpdate<Device, Index, ASSIGN> copy_functor; in Compute()
|
| D | stateful_random_ops.cc | 295 functor::DenseUpdate<Device, T, ASSIGN>()( in Compute()
|
| D | scatter_nd_op.cc | 301 functor::DenseUpdate<Device, T, ASSIGN> copy; in DoCompute()
|