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

  /external/tensorflow/tensorflow/stream_executor/cuda/
cuda_dnn.h 69 const DeviceMemory<Eigen::half>& input_h_data,
86 const DeviceMemory<float>& input_h_data,
103 const DeviceMemory<double>& input_h_data,
120 const DeviceMemory<Eigen::half>& input_h_data,
144 const DeviceMemory<float>& input_h_data,
168 const DeviceMemory<double>& input_h_data,
    [all...]
cuda_dnn.cc     [all...]
  /external/tensorflow/tensorflow/stream_executor/
dnn.h     [all...]
stream.h     [all...]
stream.cc     [all...]
  /external/tensorflow/tensorflow/contrib/cudnn_rnn/kernels/
cudnn_rnn_ops.cc 861 auto input_h_data = AsDeviceMemory<T>(input_h); variable
1061 auto input_h_data = AsDeviceMemory<T>(input_h); variable
    [all...]

Completed in 6717 milliseconds