Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/util/autotune_maps/
H A 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()
H A Dautotune_serialize.cc144 AlgorithmDesc(algorithm_config_proto.algorithm_no_scratch())) in PopulateConvMap()
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/compiler/xla/stream_executor/
Ddnn.pb.cc1259 …static const ::stream_executor::dnn::AlgorithmProto& algorithm_no_scratch(const AlgorithmConfigPro…
1267 AlgorithmConfigProto::_Internal::algorithm_no_scratch(const AlgorithmConfigProto* msg) { in algorithm_no_scratch() function in stream_executor::dnn::AlgorithmConfigProto::_Internal
1285 …:set_allocated_algorithm_no_scratch(::stream_executor::dnn::AlgorithmProto* algorithm_no_scratch) { in set_allocated_algorithm_no_scratch() argument
1288 if (algorithm_no_scratch) { in set_allocated_algorithm_no_scratch()
1290 ::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(algorithm_no_scratch); in set_allocated_algorithm_no_scratch()
1292 algorithm_no_scratch = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage( in set_allocated_algorithm_no_scratch()
1293 message_arena, algorithm_no_scratch, submessage_arena); in set_allocated_algorithm_no_scratch()
1296 _impl_.optional_algorithm_no_scratch_.algorithm_no_scratch_ = algorithm_no_scratch; in set_allocated_algorithm_no_scratch()
1525 InternalWriteMessage(2, _Internal::algorithm_no_scratch(this), in _InternalSerialize()
1526 _Internal::algorithm_no_scratch(this).GetCachedSize(), target, stream); in _InternalSerialize()
Ddnn.pb.h858 const ::stream_executor::dnn::AlgorithmProto& algorithm_no_scratch() const; in Swap()
861 …d set_allocated_algorithm_no_scratch(::stream_executor::dnn::AlgorithmProto* algorithm_no_scratch); in Swap()
867 ::stream_executor::dnn::AlgorithmProto* algorithm_no_scratch); in Swap()
1631 inline const ::stream_executor::dnn::AlgorithmProto& AlgorithmConfigProto::algorithm_no_scratch() c… in algorithm_no_scratch() function
1646 …_set_allocated_algorithm_no_scratch(::stream_executor::dnn::AlgorithmProto* algorithm_no_scratch) { in unsafe_arena_set_allocated_algorithm_no_scratch() argument
1648 if (algorithm_no_scratch) { in unsafe_arena_set_allocated_algorithm_no_scratch()
1650 _impl_.optional_algorithm_no_scratch_.algorithm_no_scratch_ = algorithm_no_scratch; in unsafe_arena_set_allocated_algorithm_no_scratch()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A 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
H A Ddnn.cc453 if (algorithm_no_scratch().has_value()) { in ToString()
454 algo_no_scratch = algorithm_no_scratch()->ToString(); in ToString()
H A Ddnn.proto147 AlgorithmProto algorithm_no_scratch = 2; field
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A 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()