Searched refs:tensor_ops_enabled (Results 1 – 12 of 12) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/tests/Dialect/lhlo_gpu/ |
H A D | lhlo_gpu_ops.mlir | 34 tensor_ops_enabled = true, 61 tensor_ops_enabled = true, 85 tensor_ops_enabled = true, 113 tensor_ops_enabled = true, 142 tensor_ops_enabled = true, 171 tensor_ops_enabled = true,
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/lhlo_to_jitrt/ |
H A D | conv_to_jitrt.mlir | 50 // CHECK-DAG: tensor_ops_enabled = false 70 tensor_ops_enabled = false, 123 tensor_ops_enabled = false, 172 tensor_ops_enabled = false, 232 tensor_ops_enabled = false, 298 tensor_ops_enabled = false,
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
H A D | gpu_conv_algorithm_picker.cc | 456 AlgorithmDesc alg_key(alg.algo_id(), alg.tensor_ops_enabled(), std::nullopt); in AutotuneOneConvRunner() 581 algo->set_tensor_ops(alg.tensor_ops_enabled()); in AutotuneOneConvRunner() 941 << " tensor_ops_enabled: " << best_algo.conv().tensor_ops_enabled(); in RunOnInstruction()
|
H A D | jitrt_custom_calls.h | 61 bool tensor_ops_enabled; member
|
H A D | jitrt_custom_calls.cc | 777 algo->set_math_type(b.tensor_ops_enabled in GetConvDescriptor()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
H A D | dnn.cc | 78 tensor_ops_enabled() == other.tensor_ops_enabled(); in operator ==() 94 if (tensor_ops_enabled()) { in ToString()
|
H A D | dnn.h | 804 bool tensor_ops_enabled() const { in tensor_ops_enabled() function
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
H A D | cuda_dnn.cc | 1294 ? algorithm_config.algorithm()->tensor_ops_enabled() in Create() 2893 if (IsTensorMathOpSet(conv) != algorithm_desc.tensor_ops_enabled()) { in AllocateCudnnConvolutionForwardWorkspace() 2942 if (IsTensorMathOpSet(conv) != algorithm_desc.tensor_ops_enabled()) { in AllocateCudnnConvolutionBackwardDataWorkspace() 2992 if (IsTensorMathOpSet(conv) != algorithm_desc.tensor_ops_enabled()) { in AllocateCudnnConvolutionBackwardFilterWorkspace() 3038 use_tensor_ops = desc->tensor_ops_enabled(); in UseTensorOps() 4065 return {{parent, cudnn, algo.algo_id(), algo.tensor_ops_enabled(), in Create() 4216 int64_t algo_id, bool tensor_ops_enabled, in CudnnLegacyConvRunner() argument 4226 tensor_ops_enabled_(tensor_ops_enabled), in CudnnLegacyConvRunner() 4827 conv.set_use_tensor_op_math(algorithm_desc.tensor_ops_enabled()); in ConvolveRunnerFromDesc() 4902 return {{parent, cudnn, algo.algo_id(), algo.tensor_ops_enabled(), in Create() [all …]
|
/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/ |
H A D | autotuning.proto | 64 bool tensor_ops_enabled = 2; field
|
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
H A D | cudnn_rnn_ops.cc | 1667 best_algo_config.algorithm()->tensor_ops_enabled(); in Compute() 1708 << algo_config->algorithm()->tensor_ops_enabled() << ")."; in MaybeAutotune() 1744 << algo.algo_id() << ", " << algo.tensor_ops_enabled() << ")."; in MaybeAutotune() 1797 << algo.algo_id() << ", " << algo.tensor_ops_enabled() << ")" in MaybeAutotune() 1811 << best_result.algorithm().tensor_ops_enabled() << ")."; in MaybeAutotune()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/ |
H A D | hlo_text_to_lhlo_no_opt.hlotxt | 299 // CHECK-SAME: tensor_ops_enabled = false 324 // CHECK-SAME: tensor_ops_enabled = false 351 // CHECK-SAME: tensor_ops_enabled = false
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/lhlo_gpu/IR/ |
H A D | lhlo_gpu_ops_enums.td | 62 BoolParameter:$tensor_ops_enabled,
|