Searched refs:conv_result_scale (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | gpu_convolution_regression_test.cc | 44 …0f_0oi->b0f, custom_call_target="__cudnn$convBackwardInput", backend_config="{conv_result_scale:1}" in TEST_F()
61 …0f_0oi->b0f, custom_call_target="__cudnn$convBackwardInput", backend_config="{conv_result_scale:1}" in TEST_F()
78 …0f_0oi->b0f, custom_call_target="__cudnn$convBackwardInput", backend_config="{conv_result_scale:1}" in TEST_F()
93 …oi01->bf01, custom_call_target="__cudnn$convBackwardFilter", backend_config="{conv_result_scale:1}" in TEST_F()
105 …s=bf01_01io->bf01, custom_call_target="__cudnn$convForward", backend_config="{conv_result_scale:1}" in TEST_F()
116 …_labels=bf_io->bf, custom_call_target="__cudnn$convForward", backend_config="{conv_result_scale:1}" in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_runner.cc | 83 if (params.config.conv_result_scale != 1) { in RunGpuConvForward()
86 params.config.conv_result_scale); in RunGpuConvForward()
130 params.config.conv_result_scale, params.config.filter_descriptor, in RunGpuConvForwardActivation()
161 if (params.config.conv_result_scale != 1) { in RunGpuConvInternalImpl()
164 params.config.conv_result_scale); in RunGpuConvInternalImpl()
173 if (params.config.conv_result_scale != 1) { in RunGpuConvInternalImpl()
176 params.config.conv_result_scale); in RunGpuConvInternalImpl()
278 config.conv_result_scale = backend_config.conv_result_scale(); in GetGpuConvConfig()
|
| D | hlo_algorithm_denylist_test.cc | 54 …1f_01io->b01f, custom_call_target="__cudnn$convForward", backend_config="{conv_result_scale:1}")"); in TEST_F()
|
| D | gpu_conv_runner.h | 68 double conv_result_scale; member
|
| D | backend_configs.proto | 31 double conv_result_scale = 4; field
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/data/ |
| D | hlo_algorithm_denylist.pbtxt | 2 …=b01f_01io->b01f, custom_call_target="__cudnn$convForward", backend_config="{conv_result_scale:1}"'
10 …=b01f_01io->b01f, custom_call_target="__cudnn$convForward", backend_config="{conv_result_scale:1}"'
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/ |
| D | conv_emitter_test.cc | 85 …s=bf01_01oi->bf01, custom_call_target="__cudnn$convForward", backend_config="{conv_result_scale:1}" in TEST()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/ |
| D | hlo_text_to_lhlo_no_opt.hlotxt | 307 …backend_config="{\"algorithm\":\"2\",\"tensor_ops_enabled\":false,\"conv_result_scale\":1,\"activa…
346 …backend_config="{\"algorithm\":\"0\",\"tensor_ops_enabled\":false,\"conv_result_scale\":1,\"activa…
386 …backend_config="{\"algorithm\":\"0\",\"tensor_ops_enabled\":false,\"conv_result_scale\":1,\"activa…
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 779 builder_.getF64FloatAttr(backend_config.conv_result_scale())); in EmitDnnConvolution()
|