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

  /external/tensorflow/tensorflow/stream_executor/cuda/
cuda_dnn.h 78 DeviceMemory<Eigen::half>* output_c_data, bool is_training,
95 DeviceMemory<float>* output_c_data, bool is_training,
112 DeviceMemory<double>* output_c_data, bool is_training,
129 const DeviceMemory<Eigen::half>& output_c_data,
153 const DeviceMemory<float>& output_c_data,
177 const DeviceMemory<double>& output_c_data,
    [all...]
cuda_dnn.cc     [all...]
  /external/tensorflow/tensorflow/contrib/cudnn_rnn/kernels/
cudnn_rnn_ops.cc 869 DeviceMemory<T> output_c_data; variable
1069 DeviceMemory<T> output_c_data; variable
    [all...]
  /external/tensorflow/tensorflow/stream_executor/
dnn.h     [all...]
stream.h     [all...]
stream.cc     [all...]

Completed in 241 milliseconds