Searched refs:conv_input_scale (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_gpu.h | 110 const se::dnn::ActivationMode activation_mode, double conv_input_scale,
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | stream.h | 369 const DeviceMemory<InputT> &conv_input_data, ScaleT conv_input_scale, in FusedConvolveWithAlgorithm() argument
385 conv_input_data, conv_input_scale, filter_descriptor, filter_data, in FusedConvolveWithAlgorithm()
413 dnn::DataType output_type, double conv_input_scale, in FusedConvolveRunnerFromDesc() argument
427 conv_input_scale, side_input_scale, leakyrelu_alpha, input_descriptor, in FusedConvolveRunnerFromDesc()
|
| D | stream_executor_pimpl.cc | 317 dnn::DataType output_type, double conv_input_scale, double side_input_scale, in GetFusedConvolveRunners() argument
332 conv_input_scale, side_input_scale, leakyrelu_alpha, stream, in GetFusedConvolveRunners()
|
| D | stream_executor_pimpl.h | 387 dnn::DataType output_type, double conv_input_scale,
|
| D | dnn.h | 1304 DeviceMemoryBase conv_input_data, double conv_input_scale, in DoFusedConvolve() argument
1413 dnn::DataType output_type, double conv_input_scale,
|
| D | dnn.cc | 145 dnn::DataType output_type, double conv_input_scale, double side_input_scale, in GetFusedConvolveRunners() argument
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_dnn.h | 351 DeviceMemoryBase conv_input_data, double conv_input_scale,
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 355 DeviceMemoryBase conv_input_data, double conv_input_scale,
|
| D | rocm_dnn.cc | 3799 DeviceMemoryBase conv_input_data, double conv_input_scale, in DoFusedConvolve() argument
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 1779 auto& conv_input_scale = op_context.op_info.inputs(4); in PredictFusedConv2DBiasActivation() local
1797 FusedChildContext(op_context, "Mul", output, {output, conv_input_scale}), in PredictFusedConv2DBiasActivation()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 3144 TRT_ShapedWeights conv_input_scale = inputs.at(4).weights(); in ConvertFusedConv2DBiasActivation() local
3145 if (conv_input_scale.count() != 1 || in ConvertFusedConv2DBiasActivation()
3146 conv_input_scale.TrtDType() != nvinfer1::DataType::kFLOAT || in ConvertFusedConv2DBiasActivation()
3147 conv_input_scale.GetSpan<float>()[0] != 1.0) { in ConvertFusedConv2DBiasActivation()
|