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

  /external/tensorflow/tensorflow/core/kernels/
cudnn_pooling_gpu.cc 218 auto output_backprop_data = local
232 output_backprop_data, &input_backprop_data)
pooling_ops_common.cc 360 auto output_backprop_data = local
374 output_backprop_data, &input_backprop_data)
  /external/tensorflow/tensorflow/stream_executor/cuda/
cuda_dnn.h 130 const DeviceMemory<Eigen::half>& output_backprop_data,
154 const DeviceMemory<float>& output_backprop_data,
178 const DeviceMemory<double>& output_backprop_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 1073 auto output_backprop_data = AsDeviceMemory<T>(output_backprop); variable
    [all...]

Completed in 128 milliseconds