Searched defs:conv_input_scale (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | stream.h | 369 const DeviceMemory<InputT> &conv_input_data, ScaleT conv_input_scale, in FusedConvolveWithAlgorithm()
413 dnn::DataType output_type, double conv_input_scale, in FusedConvolveRunnerFromDesc()
|
| H A D | dnn.cc | 145 dnn::DataType output_type, double conv_input_scale, double side_input_scale, in GetFusedConvolveRunners()
|
| H A D | stream_executor_pimpl.cc | 317 dnn::DataType output_type, double conv_input_scale, double side_input_scale, in GetFusedConvolveRunners()
|
| H A D | dnn.h | 1304 DeviceMemoryBase conv_input_data, double conv_input_scale, in DoFusedConvolve()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/ |
| H A D | op_level_cost_estimator.cc | 1779 auto& conv_input_scale = op_context.op_info.inputs(4); in PredictFusedConv2DBiasActivation() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| H A D | convert_nodes.cc | 3144 TRT_ShapedWeights conv_input_scale = inputs.at(4).weights(); in ConvertFusedConv2DBiasActivation() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A D | rocm_dnn.cc | 3799 DeviceMemoryBase conv_input_data, double conv_input_scale, in DoFusedConvolve()
|