| /external/tensorflow/tensorflow/compiler/mlir/hlo/tests/ |
| D | lhlo_gpu_ops.mlir | 57 tensor_ops_enabled = true}}
70 tensor_ops_enabled = false},
100 tensor_ops_enabled = false},
132 tensor_ops_enabled = false},
163 tensor_ops_enabled = false},
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_algorithm_picker.cc | 153 if (algo.tensor_ops_enabled()) { in AlgorithmToString()
488 result.mutable_conv()->set_tensor_ops_enabled(alg.tensor_ops_enabled()); in PickBestAlgorithmNoCacheCuda()
519 algo->set_tensor_ops(alg.tensor_ops_enabled()); in PickBestAlgorithmNoCacheCuda()
561 first_algorithm.tensor_ops_enabled()); in PickBestAlgorithmNoCacheCuda()
704 profile_result.algorithm().tensor_ops_enabled()); in PickBestAlgorithmNoCacheRocm()
742 result.mutable_conv()->set_tensor_ops_enabled(alg.tensor_ops_enabled()); in PickBestAlgorithmNoCacheRocm()
784 << " tensor_ops_enabled: " << best_algo.conv().tensor_ops_enabled(); in RunOnInstruction()
796 backend_config.set_tensor_ops_enabled(best_algo.conv().tensor_ops_enabled()); in RunOnInstruction()
|
| D | backend_configs.proto | 28 bool tensor_ops_enabled = 2; field
|
| D | gpu_conv_runner.cc | 276 backend_config.tensor_ops_enabled()), in GetGpuConvConfig()
319 << config.algorithm.algorithm()->tensor_ops_enabled(); in GetGpuConvConfig()
|
| D | ir_emitter_unnested.cc | 1181 op.backend_config().tensor_ops_enabled().getValue()); in EmitConvolutionThunkFromMlir()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gpu_utils.cc | 274 selected_result->conv().tensor_ops_enabled()}); in BestCudnnConvAlgorithm()
279 selected_result_no_scratch->conv().tensor_ops_enabled()}); in BestCudnnConvAlgorithm()
|
| D | conv_grad_input_ops.cc | 429 profile_algorithm.tensor_ops_enabled()); in operator ()()
464 profile_result.algorithm().tensor_ops_enabled()); in operator ()()
484 profile_algorithm.tensor_ops_enabled()); in operator ()()
|
| D | conv_ops_3d.cc | 550 profile_algorithm.tensor_ops_enabled()); in launch()
585 profile_result.algorithm().tensor_ops_enabled()); in launch()
606 profile_algorithm.tensor_ops_enabled()); in launch()
|
| D | conv_ops.cc | 1036 profile_algorithm.tensor_ops_enabled()); in operator ()()
1074 profile_result.algorithm().tensor_ops_enabled()); in operator ()()
1093 profile_algorithm.tensor_ops_enabled()); in operator ()()
1113 << algorithm_config.algorithm()->tensor_ops_enabled(); in operator ()()
|
| D | conv_grad_filter_ops.cc | 1024 profile_algorithm.tensor_ops_enabled()); in operator ()()
1062 profile_result.algorithm().tensor_ops_enabled()); in operator ()()
1082 profile_algorithm.tensor_ops_enabled()); in operator ()()
|
| D | cudnn_rnn_ops.cc | 1624 best_algo_config.algorithm()->tensor_ops_enabled(); in Compute()
1665 << algo_config->algorithm()->tensor_ops_enabled() << ")."; in MaybeAutoTune()
1700 << algo.algo_id() << ", " << algo.tensor_ops_enabled() << ")."; in MaybeAutoTune()
1753 << algo.algo_id() << ", " << algo.tensor_ops_enabled() << ")" in MaybeAutoTune()
1767 << best_result.algorithm().tensor_ops_enabled() << ")."; in MaybeAutoTune()
|
| D | conv_grad_ops_3d.cc | 1464 profile_algorithm.tensor_ops_enabled()); in Compute()
1503 profile_algorithm.tensor_ops_enabled()); in Compute()
1948 profile_algorithm.tensor_ops_enabled()); in Compute()
1980 profile_algorithm.tensor_ops_enabled()); in Compute()
|
| D | conv_ops_fused_impl.h | 414 profile_algorithm.tensor_ops_enabled());
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/ |
| D | hlo_text_to_lhlo_no_opt.hlotxt | 282 // CHECK-SAME: tensor_ops_enabled = false
307 …backend_config="{\"algorithm\":\"2\",\"tensor_ops_enabled\":false,\"conv_result_scale\":1,\"activa…
319 // CHECK-SAME: tensor_ops_enabled = false
346 …convBiasActivationForward", backend_config="{\"algorithm\":\"0\",\"tensor_ops_enabled\":false,\"co…
358 // CHECK-SAME: tensor_ops_enabled = false
386 …convBiasActivationForward", backend_config="{\"algorithm\":\"0\",\"tensor_ops_enabled\":false,\"co…
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | autotuning.proto | 53 bool tensor_ops_enabled = 2; field
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | lhlo_gpu_ops_structs.td | 24 StructFieldAttr<"tensor_ops_enabled", BoolAttr>,
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.cc | 32 auto p = std::make_pair(algo_id(), tensor_ops_enabled()); in hash()
37 if (tensor_ops_enabled()) { in ToString()
|
| D | dnn.h | 751 bool tensor_ops_enabled() const { in tensor_ops_enabled() function
757 tensor_ops_enabled() == other.tensor_ops_enabled();
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1119 ? algorithm_config.algorithm()->tensor_ops_enabled() in Create()
2541 if (IsTensorMathOpSet(conv) != algorithm_desc.tensor_ops_enabled()) { in AllocateCudnnConvolutionForwardWorkspace()
2585 if (IsTensorMathOpSet(conv) != algorithm_desc.tensor_ops_enabled()) { in AllocateCudnnConvolutionBackwardDataWorkspace()
2631 if (IsTensorMathOpSet(conv) != algorithm_desc.tensor_ops_enabled()) { in AllocateCudnnConvolutionBackwardFilterWorkspace()
2673 use_tensor_ops = desc->tensor_ops_enabled(); in UseTensorOps()
3223 conv.set_use_tensor_op_math(algo_desc.tensor_ops_enabled()); in DoFusedConvolveImpl()
3262 if (IsTensorMathOpSet(conv) != algo_desc.tensor_ops_enabled()) { in DoFusedConvolveImpl()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 782 builder_.getBoolAttr(backend_config.tensor_ops_enabled()), in EmitDnnConvolution()
|