Home
last modified time | relevance | path

Searched refs:output_c_data (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h87 DeviceMemory<Eigen::half>* output_c_data, bool is_training,
105 DeviceMemory<float>* output_c_data, bool is_training,
123 DeviceMemory<double>* output_c_data, bool is_training,
141 const DeviceMemory<Eigen::half>& output_c_data,
166 const DeviceMemory<float>& output_c_data,
191 const DeviceMemory<double>& output_c_data,
653 DeviceMemory<T>* output_c_data, bool is_training,
672 const DeviceMemory<T>& output_c_data,
Dcuda_dnn.cc1610 const DeviceMemory<T>& output_c_data) { in ExtractAndCheckRnnForward() argument
1759 DeviceMemory<T>* output_c_data, bool is_training, in DoRnnForwardImpl() argument
1768 output_h_desc, *output_h_data, output_c_desc, *output_c_data)); in DoRnnForwardImpl()
1816 /*cyDesc=*/output_c_desc.handle(), /*cy=*/output_c_data->opaque(), in DoRnnForwardImpl()
1832 /*cy=*/output_c_data->opaque(), /*workspace=*/workspace.opaque(), in DoRnnForwardImpl()
1847 /*cyDesc=*/output_c_desc.handle(), /*cy=*/output_c_data->opaque(), in DoRnnForwardImpl()
1865 /*cy=*/output_c_data->opaque(), /*workspace=*/workspace.opaque(), in DoRnnForwardImpl()
1899 const DeviceMemory<T>& output_c_data, in DoRnnBackwardImpl() argument
1915 output_h_data, output_c_desc, output_c_data)); in DoRnnBackwardImpl()
2140 DeviceMemory<Eigen::half>* output_c_data, bool is_training, in DoRnnForward() argument
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h114 DeviceMemory<Eigen::half>* output_c_data, bool is_training,
132 DeviceMemory<float>* output_c_data, bool is_training,
150 DeviceMemory<double>* output_c_data, bool is_training,
168 const DeviceMemory<Eigen::half>& output_c_data,
193 const DeviceMemory<float>& output_c_data,
218 const DeviceMemory<double>& output_c_data,
751 DeviceMemory<T>* output_c_data, bool is_training,
768 const DeviceMemory<T>& output_c_data,
Drocm_dnn.cc1985 const DeviceMemory<T>& output_c_data, RnnModelDims* model_dims) { in ExtractAndCheckRnnForward() argument
2095 DeviceMemory<T>* output_c_data, bool is_training, in DoRnnForwardImpl() argument
2103 output_h_desc, *output_h_data, output_c_desc, *output_c_data, in DoRnnForwardImpl()
2165 output_c_data->opaque() /*cy*/, workspace.opaque() /*workspace*/, in DoRnnForwardImpl()
2183 output_c_data->opaque() /*cy*/, workspace.opaque() /*workspace*/, in DoRnnForwardImpl()
2210 const DeviceMemory<T>& output_c_data, in DoRnnBackwardImpl() argument
2225 output_h_desc, output_h_data, output_c_desc, output_c_data, &model_dims); in DoRnnBackwardImpl()
2591 DeviceMemory<Eigen::half>* output_c_data, bool is_training, in DoRnnForward() argument
2616 output_h_data, miopen_output_c_desc, output_c_data, is_training, in DoRnnForward()
2633 DeviceMemory<float>* output_c_data, bool is_training, in DoRnnForward() argument
[all …]
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h2198 DeviceMemory<Eigen::half>* output_c_data, in DoRnnForward() argument
2219 DeviceMemory<float>* output_c_data, in DoRnnForward() argument
2240 DeviceMemory<double>* output_c_data, in DoRnnForward() argument
2302 const DeviceMemory<Eigen::half>& output_c_data, in DoRnnBackward() argument
2330 const DeviceMemory<float>& output_c_data, in DoRnnBackward() argument
2358 const DeviceMemory<double>& output_c_data, in DoRnnBackward() argument
Dstream.h1792 DeviceMemory<Eigen::half> *output_c_data,
1811 DeviceMemory<float> *output_c_data, bool is_training,
1829 DeviceMemory<double> *output_c_data, bool is_training,
1850 const DeviceMemory<Eigen::half> &output_c_data,
1875 const DeviceMemory<float> &output_c_data,
1900 const DeviceMemory<double> &output_c_data,
Dstream.cc4552 DeviceMemory<Eigen::half> *output_c_data, bool is_training, in ThenRnnForward() argument
4561 output_h_desc, output_h_data, output_c_desc, output_c_data, is_training, in ThenRnnForward()
4585 DeviceMemory<float> *output_c_data, bool is_training, in ThenRnnForward() argument
4594 output_h_desc, output_h_data, output_c_desc, output_c_data, is_training, in ThenRnnForward()
4619 DeviceMemory<double> *output_c_data, bool is_training, in ThenRnnForward() argument
4628 output_h_desc, output_h_data, output_c_desc, output_c_data, is_training, in ThenRnnForward()
4653 const DeviceMemory<Eigen::half> &output_c_data, in ThenRnnBackward() argument
4669 output_h_desc, output_h_data, output_c_desc, output_c_data, in ThenRnnBackward()
4697 const DeviceMemory<float> &output_c_data, in ThenRnnBackward() argument
4713 output_h_desc, output_h_data, output_c_desc, output_c_data, in ThenRnnBackward()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc823 DeviceMemory<T> output_c_data; in DoForward() local
825 output_c_data = AsDeviceMemory<T>(output_c); in DoForward()
835 &output_c_data, is_training, reserve_space_allocator, in DoForward()
885 DeviceMemory<T> output_c_data; in DoBackward() local
887 output_c_data = AsDeviceMemory<T>(output_c); in DoBackward()
914 output_c_data, output_backprop_data, output_h_backprop_data, in DoBackward()