Searched defs:side_input_scale (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | backend_configs.proto | 42 double side_input_scale = 5; field
|
| D | gpu_conv_runner.h | 61 double side_input_scale; member
|
| /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 | 393 const DeviceMemory<SideInputT> &side_input_data, ScaleT side_input_scale, in FusedConvolveWithAlgorithm()
421 const DeviceMemory<SideInputT> &side_input_data, ScaleT side_input_scale, in FusedConvolveWithExecutionPlan()
|
| D | dnn.h | 1260 DeviceMemoryBase side_input_data, double side_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()
4875 DeviceMemoryBase side_input_data, double side_input_scale, in DoFusedConvolve()
5022 DeviceMemoryBase side_input_data, double side_input_scale, in DoFusedConvolveWithExecutionPlan()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 1759 auto& side_input_scale = op_context.op_info.inputs(5); in PredictFusedConv2DBiasActivation() local
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 3678 DeviceMemoryBase side_input_data, double side_input_scale, in DoFusedConvolve()
|