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

  /external/tensorflow/tensorflow/stream_executor/cuda/
cuda_dnn.h 145 DeviceMemory<Eigen::half>* input_h_backprop_data,
170 DeviceMemory<float>* input_h_backprop_data,
195 DeviceMemory<double>* input_h_backprop_data,
    [all...]
cuda_dnn.cc     [all...]
  /external/tensorflow/tensorflow/stream_executor/rocm/
rocm_dnn.h 136 DeviceMemory<Eigen::half>* input_h_backprop_data,
161 DeviceMemory<float>* input_h_backprop_data,
186 DeviceMemory<double>* input_h_backprop_data,
    [all...]
rocm_dnn.cc     [all...]
  /external/tensorflow/tensorflow/stream_executor/
dnn.h     [all...]
stream.h     [all...]
stream.cc     [all...]
  /external/tensorflow/tensorflow/core/kernels/
cudnn_rnn_ops.cc 836 auto input_h_backprop_data = AsDeviceMemory<T>(input_h_backprop); local
856 &input_h_backprop_data, &input_c_backprop_data,
    [all...]

Completed in 244 milliseconds