Home
last modified time | relevance | path

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 Dlhlo_gpu_ops.mlir34 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 Dconv_to_jitrt.mlir50 // 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 Dgpu_conv_algorithm_picker.cc456 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 Djitrt_custom_calls.h61 bool tensor_ops_enabled; member
H A Djitrt_custom_calls.cc777 algo->set_math_type(b.tensor_ops_enabled in GetConvDescriptor()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Ddnn.cc78 tensor_ops_enabled() == other.tensor_ops_enabled(); in operator ==()
94 if (tensor_ops_enabled()) { in ToString()
H A Ddnn.h804 bool tensor_ops_enabled() const { in tensor_ops_enabled() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_dnn.cc1294 ? 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 Dautotuning.proto64 bool tensor_ops_enabled = 2; field
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dcudnn_rnn_ops.cc1667 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 Dhlo_text_to_lhlo_no_opt.hlotxt299 // 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 Dlhlo_gpu_ops_enums.td62 BoolParameter:$tensor_ops_enabled,