HomeSort by relevance Sort by last modified time
    Searched refs:TensorZero (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/contrib/rnn/kernels/
lstm_ops.cc 637 functor::TensorZero<Device, T>()(device, wci_grad_tensor->flat<T>());
638 functor::TensorZero<Device, T>()(device, wcf_grad_tensor->flat<T>());
639 functor::TensorZero<Device, T>()(device, wco_grad_tensor->flat<T>());
    [all...]
lstm_ops_gpu.cu.cc 62 struct TensorZero<GPUDevice, T> {
403 template struct TensorZero<GPUDevice, T>; \
lstm_ops.h 31 struct TensorZero {

Completed in 2273 milliseconds