Home
last modified time | relevance | path

Searched refs:use_tensor_ops (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_dnn.cc1292 bool use_tensor_ops = in Create() local
1296 if (use_tensor_ops && !allow_tensor_ops) { in Create()
1303 use_tensor_ops ? CUDNN_TENSOR_OP_MATH : CUDNN_FMA_MATH; in Create()
1306 use_tensor_ops ? CUDNN_TENSOR_OP_MATH : CUDNN_DEFAULT_MATH; in Create()
3036 bool use_tensor_ops; in UseTensorOps() local
3038 use_tensor_ops = desc->tensor_ops_enabled(); in UseTensorOps()
3039 if (use_tensor_ops && !IsTensorMathEnabled(stream, type)) { in UseTensorOps()
3044 use_tensor_ops = IsTensorMathEnabled(stream, type); in UseTensorOps()
3046 return use_tensor_ops; in UseTensorOps()
3065 bool use_tensor_ops; in GetCudnnConvolutionForwardAlgorithm() local
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Ddnn.h787 AlgorithmDesc(Index algo_id, bool use_tensor_ops) in AlgorithmDesc() argument
788 : AlgorithmDesc(algo_id, use_tensor_ops, std::nullopt) {} in AlgorithmDesc()
789 AlgorithmDesc(Index algo_id, bool use_tensor_ops, in AlgorithmDesc() argument
793 proto_.set_math_type(use_tensor_ops ? AlgorithmProto::TENSOR_OP_MATH in AlgorithmDesc()