Searched defs:conv_input_scale (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.h | 381 double conv_input_scale, double side_input_scale, Stream *stream, in GetFusedConvolveExecutionPlans()
|
| D | stream.h | 389 const DeviceMemory<InputT> &conv_input_data, ScaleT conv_input_scale, in FusedConvolveWithAlgorithm()
417 const DeviceMemory<InputT> &conv_input_data, ScaleT conv_input_scale, in FusedConvolveWithExecutionPlan()
|
| D | dnn.h | 1256 DeviceMemoryBase conv_input_data, double conv_input_scale, in DoFusedConvolve()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 4302 double conv_input_scale, double side_input_scale, Stream* stream, in GetFusedConvolveExecutionPlans()
4871 DeviceMemoryBase conv_input_data, double conv_input_scale, in DoFusedConvolve()
5018 DeviceMemoryBase conv_input_data, double conv_input_scale, in DoFusedConvolveWithExecutionPlan()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 1758 auto& conv_input_scale = op_context.op_info.inputs(4); in PredictFusedConv2DBiasActivation() local
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 3674 DeviceMemoryBase conv_input_data, double conv_input_scale, in DoFusedConvolve()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 3399 TRT_ShapedWeights conv_input_scale = inputs.at(4).weights(); in ConvertFusedConv2DBiasActivation() local
|