Home
last modified time | relevance | path

Searched refs:side_input_scale (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/contrib/fused_conv/python/ops/
Dfused_conv2d_bias_activation_op.py36 side_input_scale=0.0, argument
104 side_input_scale,
Dfused_conv2d_bias_activation_op_test_base.py671 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/
Dfused_conv2d_bias_activation_op.h43 ScaleType side_input_scale, const Tensor& bias,
57 ScaleType side_input_scale, const Tensor& bias,
Dfused_conv2d_bias_activation_op.cc191 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/
Dcudnn_conv_runner.cc57 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()
Dbackend_configs.proto40 double side_input_scale = 5; field
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h279 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,
Dcuda_dnn.cc2929 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/
Drocm_dnn.h270 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,
Drocm_dnn.cc3151 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/
Ddnn.h1121 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,
Dstream.cc446 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 …]
Dstream.h343 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/
Dop_level_cost_estimator.cc1179 auto& side_input_scale = op_context.op_info.inputs(5); in PredictFusedConv2DBiasActivation() local
1207 {side_input, side_input_scale})); in PredictFusedConv2DBiasActivation()