Home
last modified time | relevance | path

Searched defs:side_input_scale (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/contrib/fused_conv/python/ops/
Dfused_conv2d_bias_activation_op.py36 side_input_scale=0.0, argument
Dfused_conv2d_bias_activation_op_test.py624 padding, strides, side_input_scale, argument
/external/tensorflow/tensorflow/contrib/fused_conv/kernels/
Dfused_conv2d_bias_activation_op.cc181 auto side_input_scale = *reinterpret_cast<const ScaleType*>( in Compute() local
310 ScaleType side_input_scale, const Tensor& bias, in launch()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc2508 const DeviceMemory<Type>& side_input_data, ScaleType side_input_scale, in DoFusedConvolveImpl()
2974 const DeviceMemory<double>& side_input_data, double side_input_scale, in DoFusedConvolve()
2996 const DeviceMemory<float>& side_input_data, float side_input_scale, in DoFusedConvolve()
3018 const DeviceMemory<Eigen::half>& side_input_data, float side_input_scale, in DoFusedConvolve()
3041 const DeviceMemory<int8>& side_input_data, float side_input_scale, in DoFusedConvolve()
/external/tensorflow/tensorflow/stream_executor/
Dstream.cc421 const DeviceMemory<int8> &side_input_data, float side_input_scale, in ThenFusedConvolveWithScratch()
454 const DeviceMemory<Eigen::half> &side_input_data, float side_input_scale, in ThenFusedConvolveWithScratch()
488 const DeviceMemory<float> &side_input_data, float side_input_scale, in ThenFusedConvolveWithScratch()
575 const DeviceMemory<float> &side_input_data, float side_input_scale, in ThenFusedConvolveWithAlgorithm()
613 const DeviceMemory<Eigen::half> &side_input_data, float side_input_scale, in ThenFusedConvolveWithAlgorithm()
652 const DeviceMemory<int8> &side_input_data, float side_input_scale, in ThenFusedConvolveWithAlgorithm()
752 const DeviceMemory<int8> &side_input_data, float side_input_scale, in ThenFusedConvolve()