Searched defs:conv_input_scale (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 3216 ScaleType conv_input_scale, const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolveImpl()
3780 const DeviceMemory<double>& conv_input_data, double conv_input_scale, in DoFusedConvolve()
3804 const DeviceMemory<float>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
3828 const DeviceMemory<Eigen::half>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
3853 const DeviceMemory<int8>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
3886 const DeviceMemory<int8>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 3436 const DeviceMemory<double>& conv_input_data, double conv_input_scale, in DoFusedConvolve()
3453 const DeviceMemory<float>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
3470 const DeviceMemory<Eigen::half>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
3488 const DeviceMemory<int8>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 1317 auto& conv_input_scale = op_context.op_info.inputs(4); in PredictFusedConv2DBiasActivation() local
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 462 const DeviceMemory<double> &conv_input_data, double conv_input_scale, in ThenFusedConvolveWithAlgorithm()
500 const DeviceMemory<float> &conv_input_data, float conv_input_scale, in ThenFusedConvolveWithAlgorithm()
538 const DeviceMemory<Eigen::half> &conv_input_data, float conv_input_scale, in ThenFusedConvolveWithAlgorithm()
577 const DeviceMemory<int8> &conv_input_data, float conv_input_scale, in ThenFusedConvolveWithAlgorithm()
615 const DeviceMemory<int8> &conv_input_data, float conv_input_scale, in ThenFusedConvolveWithAlgorithm()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 3170 TRT_ShapedWeights conv_input_scale = inputs.at(4).weights(); in ConvertFusedConv2DBiasActivation() local
|