Searched defs:conv_input_data (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.h | 389 const DeviceMemory<InputT> &conv_input_data, ScaleT conv_input_scale, in FusedConvolveWithAlgorithm()
417 const DeviceMemory<InputT> &conv_input_data, ScaleT conv_input_scale, in FusedConvolveWithExecutionPlan()
|
| D | dnn.h | 1256 DeviceMemoryBase conv_input_data, double conv_input_scale, in DoFusedConvolve()
2397 const DeviceMemory<float>& conv_input_data, in DoFusedConvolutionBiasActivation()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 3674 DeviceMemoryBase conv_input_data, double conv_input_scale, in DoFusedConvolve()
4510 const DeviceMemory<T>& conv_input_data, in DoFusedConvolutionBiasActivationImpl()
4598 const DeviceMemory<float>& conv_input_data, in DoFusedConvolutionBiasActivation()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 4871 DeviceMemoryBase conv_input_data, double conv_input_scale, in DoFusedConvolve()
5018 DeviceMemoryBase conv_input_data, double conv_input_scale, in DoFusedConvolveWithExecutionPlan()
|