Searched defs:side_input_scale (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | backend_configs.proto | 40 double side_input_scale = 5; field
|
| D | cudnn_conv_runner.cc | 57 double side_input_scale; member
|
| /external/tensorflow/tensorflow/contrib/fused_conv/python/ops/ |
| D | fused_conv2d_bias_activation_op.py | 36 side_input_scale=0.0, argument
|
| D | fused_conv2d_bias_activation_op_test_base.py | 671 padding, strides, side_input_scale, argument
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.cc | 191 auto side_input_scale = *reinterpret_cast<const ScaleType*>( in Compute() local
445 ScaleType side_input_scale, const Tensor& bias, in launch()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 2929 ScaleType side_input_scale, const dnn::BatchDescriptor& bias_descriptor, in DoFusedConvolveImpl()
3349 const DeviceMemory<double>& side_input_data, double side_input_scale, in DoFusedConvolve()
3373 const DeviceMemory<float>& side_input_data, float side_input_scale, in DoFusedConvolve()
3397 const DeviceMemory<Eigen::half>& side_input_data, float side_input_scale, in DoFusedConvolve()
3422 const DeviceMemory<int8>& side_input_data, float side_input_scale, in DoFusedConvolve()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 3151 const DeviceMemory<double>& side_input_data, double side_input_scale, in DoFusedConvolve()
3168 const DeviceMemory<float>& side_input_data, float side_input_scale, in DoFusedConvolve()
3185 const DeviceMemory<Eigen::half>& side_input_data, float side_input_scale, in DoFusedConvolve()
3203 const DeviceMemory<int8>& side_input_data, float side_input_scale, in DoFusedConvolve()
|
| /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
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 446 const DeviceMemory<double> &side_input_data, double side_input_scale, in ThenFusedConvolveWithAlgorithm()
484 const DeviceMemory<float> &side_input_data, float side_input_scale, in ThenFusedConvolveWithAlgorithm()
522 const DeviceMemory<Eigen::half> &side_input_data, float side_input_scale, in ThenFusedConvolveWithAlgorithm()
561 const DeviceMemory<int8> &side_input_data, float side_input_scale, in ThenFusedConvolveWithAlgorithm()
|