/external/tensorflow/tensorflow/contrib/fused_conv/python/ops/ |
D | fused_conv2d_bias_activation_op.py | 36 side_input_scale=0.0, argument 104 side_input_scale,
|
D | fused_conv2d_bias_activation_op_test_base.py | 671 padding, strides, side_input_scale, argument 699 conv_and_side_inputs = conv_result + side_input_scale * _NchwVectCToNchw( 868 side_input_scale = test_param["side_input_scale"] 920 side_input_scale=side_input_scale, 928 side_input_scale, side_input, biases, apply_relu)
|
/external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
D | fused_conv2d_bias_activation_op.h | 43 ScaleType side_input_scale, const Tensor& bias, 57 ScaleType side_input_scale, const Tensor& bias,
|
D | fused_conv2d_bias_activation_op.cc | 191 auto side_input_scale = *reinterpret_cast<const ScaleType*>( in Compute() local 197 if (side_input_scale != 0) { in Compute() 242 side_input, side_input_scale, bias, activation_mode_, in Compute() 445 ScaleType side_input_scale, const Tensor& bias, in launch() argument 555 if (side_input_scale != 0) { in launch() 676 (side_input_scale != 0), in launch() 725 side_input_scale, bias_desc, bias_ptr, dnn_activation_mode, in launch() 758 filter_ptr, conv_desc, side_input_ptr, side_input_scale, in launch()
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | cudnn_conv_runner.cc | 57 double side_input_scale; member 284 if (params.fusion->side_input_scale != 0) { in RunCudnnConvImpl() 301 params.fusion->side_input_scale, bias_desc, in RunCudnnConvImpl() 379 fusion.side_input_scale = backend_config.side_input_scale(); in GetCudnnConvParams()
|
D | backend_configs.proto | 40 double side_input_scale = 5; field
|
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_dnn.h | 279 const DeviceMemory<double>& side_input_data, double side_input_scale, 293 const DeviceMemory<float>& side_input_data, float side_input_scale, 309 float side_input_scale, 325 const DeviceMemory<int8>& side_input_data, float side_input_scale, 590 ScaleType side_input_scale, const dnn::BatchDescriptor& bias_descriptor,
|
D | cuda_dnn.cc | 2929 ScaleType side_input_scale, const dnn::BatchDescriptor& bias_descriptor, in DoFusedConvolveImpl() argument 2983 auto side_input_data_ptr = (side_input_scale == 0) ? output_data->opaque() in DoFusedConvolveImpl() 2995 << "\nside_input_scale = " << side_input_scale in DoFusedConvolveImpl() 3019 /*workSpaceSizeInBytes=*/scratch.size(), /*alpha2=*/&side_input_scale, in DoFusedConvolveImpl() 3349 const DeviceMemory<double>& side_input_data, double side_input_scale, in DoFusedConvolve() argument 3360 side_input_data, side_input_scale, bias_descriptor, biases, in DoFusedConvolve() 3373 const DeviceMemory<float>& side_input_data, float side_input_scale, in DoFusedConvolve() argument 3384 side_input_data, side_input_scale, bias_descriptor, biases, in DoFusedConvolve() 3397 const DeviceMemory<Eigen::half>& side_input_data, float side_input_scale, in DoFusedConvolve() argument 3409 side_input_data, side_input_scale, bias_descriptor, biases, in DoFusedConvolve() [all …]
|
/external/tensorflow/tensorflow/stream_executor/rocm/ |
D | rocm_dnn.h | 270 const DeviceMemory<double>& side_input_data, double side_input_scale, 284 const DeviceMemory<float>& side_input_data, float side_input_scale, 300 float side_input_scale, 316 const DeviceMemory<int8>& side_input_data, float side_input_scale,
|
D | rocm_dnn.cc | 3151 const DeviceMemory<double>& side_input_data, double side_input_scale, in DoFusedConvolve() argument 3168 const DeviceMemory<float>& side_input_data, float side_input_scale, in DoFusedConvolve() argument 3185 const DeviceMemory<Eigen::half>& side_input_data, float side_input_scale, in DoFusedConvolve() argument 3203 const DeviceMemory<int8>& side_input_data, float side_input_scale, in DoFusedConvolve() argument
|
/external/tensorflow/tensorflow/stream_executor/ |
D | dnn.h | 1121 const DeviceMemory<double>& side_input_data, double side_input_scale, 1138 const DeviceMemory<float>& side_input_data, float side_input_scale, 1156 const DeviceMemory<Eigen::half>& side_input_data, float side_input_scale, 1176 const DeviceMemory<int8>& side_input_data, float side_input_scale,
|
D | stream.cc | 446 const DeviceMemory<double> &side_input_data, double side_input_scale, in ThenFusedConvolveWithAlgorithm() argument 456 PARAM(side_input_data), PARAM(side_input_scale), in ThenFusedConvolveWithAlgorithm() 465 side_input_data, side_input_scale, bias_descriptor, biases, in ThenFusedConvolveWithAlgorithm() 484 const DeviceMemory<float> &side_input_data, float side_input_scale, in ThenFusedConvolveWithAlgorithm() argument 494 PARAM(side_input_data), PARAM(side_input_scale), in ThenFusedConvolveWithAlgorithm() 503 side_input_data, side_input_scale, bias_descriptor, biases, in ThenFusedConvolveWithAlgorithm() 522 const DeviceMemory<Eigen::half> &side_input_data, float side_input_scale, in ThenFusedConvolveWithAlgorithm() argument 533 PARAM(side_input_data), PARAM(side_input_scale), in ThenFusedConvolveWithAlgorithm() 542 side_input_data, side_input_scale, bias_descriptor, biases, in ThenFusedConvolveWithAlgorithm() 561 const DeviceMemory<int8> &side_input_data, float side_input_scale, in ThenFusedConvolveWithAlgorithm() argument [all …]
|
D | stream.h | 343 const DeviceMemory<double> &side_input_data, double side_input_scale, 357 const DeviceMemory<float> &side_input_data, float side_input_scale, 371 const DeviceMemory<Eigen::half> &side_input_data, float side_input_scale, 386 const DeviceMemory<int8> &side_input_data, float side_input_scale,
|
/external/tensorflow/tensorflow/core/grappler/costs/ |
D | op_level_cost_estimator.cc | 1179 auto& side_input_scale = op_context.op_info.inputs(5); in PredictFusedConv2DBiasActivation() local 1207 {side_input, side_input_scale})); in PredictFusedConv2DBiasActivation()
|