Home
last modified time | relevance | path

Searched refs:algorithm_no_scratch (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/util/autotune_maps/
Dautotune_serialize_test.cc108 AlgorithmDesc algorithm_no_scratch(/*algo_id=*/1, /*use_tensor_ops=*/true); in TEST() local
109 AutotuneEntry<se::dnn::ConvOp> example_a(algorithm, algorithm_no_scratch); in TEST()
158 AlgorithmDesc algorithm_no_scratch(/*algo_id=*/1, /*use_tensor_ops=*/true); in TEST() local
160 algorithm_no_scratch); in TEST()
Dautotune_serialize.cc144 AlgorithmDesc(algorithm_config_proto.algorithm_no_scratch())) in PopulateConvMap()
/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Ddnn.h930 AlgorithmConfig(AlgorithmDesc algorithm, AlgorithmDesc algorithm_no_scratch) in AlgorithmConfig() argument
931 : algorithm_(algorithm), algorithm_no_scratch_(algorithm_no_scratch) {} in AlgorithmConfig()
933 AlgorithmDesc algorithm_no_scratch) in AlgorithmConfig() argument
935 algorithm_no_scratch_(algorithm_no_scratch), in AlgorithmConfig()
950 algorithm_config_proto.algorithm_no_scratch(); in AlgorithmConfig()
957 std::optional<AlgorithmDesc> algorithm_no_scratch() const { in algorithm_no_scratch() function
Ddnn.proto147 AlgorithmProto algorithm_no_scratch = 2; field
Ddnn.cc453 if (algorithm_no_scratch().has_value()) { in ToString()
454 algo_no_scratch = algorithm_no_scratch()->ToString(); in ToString()
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_dnn.cc3094 algo_desc = algorithm_config.algorithm_no_scratch(); in GetCudnnConvolutionForwardAlgorithm()
3156 algo_desc = algorithm_config.algorithm_no_scratch(); in GetCudnnConvolutionBackwardDataAlgorithm()
3218 algo_desc = algorithm_config.algorithm_no_scratch(); in GetCudnnConvolutionBackwardFilterAlgorithm()