Home
last modified time | relevance | path

Searched defs: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
3036 bool use_tensor_ops; in UseTensorOps() local
3065 bool use_tensor_ops; in GetCudnnConvolutionForwardAlgorithm() local
3127 bool use_tensor_ops; in GetCudnnConvolutionBackwardDataAlgorithm() local
3188 bool use_tensor_ops; in GetCudnnConvolutionBackwardFilterAlgorithm() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Ddnn.h787 AlgorithmDesc(Index algo_id, bool use_tensor_ops) in AlgorithmDesc()
789 AlgorithmDesc(Index algo_id, bool use_tensor_ops, in AlgorithmDesc()