Home
last modified time | relevance | path

Searched refs:TensorZero (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/contrib/rnn/kernels/
Dlstm_ops.cc637 functor::TensorZero<Device, T>()(device, wci_grad_tensor->flat<T>()); in Compute()
638 functor::TensorZero<Device, T>()(device, wcf_grad_tensor->flat<T>()); in Compute()
639 functor::TensorZero<Device, T>()(device, wco_grad_tensor->flat<T>()); in Compute()
1019 void TensorZero<GPUDevice, T>::operator()(const GPUDevice& d, \
1022 extern template struct TensorZero<GPUDevice, T>; \
1207 functor::TensorZero<Device, T>()(device, cs_grad_tensor.flat<T>()); in Compute()
1208 functor::TensorZero<Device, T>()(device, cs_prev_grad_tensor->flat<T>()); in Compute()
1209 functor::TensorZero<Device, T>()(device, h_grad_tensor.flat<T>()); in Compute()
1210 functor::TensorZero<Device, T>()(device, h_prev_grad_tensor->flat<T>()); in Compute()
1211 functor::TensorZero<Device, T>()(device, w_grad_tensor->flat<T>()); in Compute()
[all …]
Dlstm_ops_gpu.cu.cc62 struct TensorZero<GPUDevice, T> { struct
403 template struct TensorZero<GPUDevice, T>; \
Dlstm_ops.h31 struct TensorZero { struct