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.cc105 AlgorithmDesc algorithm_no_scratch(/*algo_id=*/1, /*use_tensor_op=*/true); in TEST() local
107 algorithm_no_scratch); in TEST()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h874 AlgorithmConfig(AlgorithmDesc algorithm, AlgorithmDesc algorithm_no_scratch) in AlgorithmConfig() argument
875 : algorithm_(algorithm), algorithm_no_scratch_(algorithm_no_scratch) {} in AlgorithmConfig()
877 AlgorithmDesc algorithm_no_scratch) in AlgorithmConfig() argument
879 algorithm_no_scratch_(algorithm_no_scratch), in AlgorithmConfig()
896 algorithm_config_proto.algorithm_no_scratch(); in AlgorithmConfig()
906 absl::optional<AlgorithmDesc> algorithm_no_scratch() const { in algorithm_no_scratch() function
Ddnn.proto124 AlgorithmProto algorithm_no_scratch = 2; field
Ddnn.cc349 if (algorithm_no_scratch().has_value()) { in ToString()
350 algo_no_scratch = algorithm_no_scratch()->ToString(); in ToString()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_runner.cc261 algorithm.algorithm_no_scratch().has_value() in RunGpuConvImpl()
262 ? absl::StrCat(algorithm.algorithm_no_scratch()->algo_id()) in RunGpuConvImpl()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc3021 algo_desc = algorithm_config.algorithm_no_scratch(); in GetCudnnConvolutionForwardAlgorithm()
3083 algo_desc = algorithm_config.algorithm_no_scratch(); in GetCudnnConvolutionBackwardDataAlgorithm()
3145 algo_desc = algorithm_config.algorithm_no_scratch(); in GetCudnnConvolutionBackwardFilterAlgorithm()
4075 plan_config.algorithm_no_scratch(); in DoConvolveWithExecutionPlan()
5034 algorithm_config.algorithm_no_scratch(); in DoFusedConvolveWithExecutionPlan()