Searched refs:algorithm_config_proto (Results 1 – 2 of 2) sorted by relevance
139 const AlgorithmConfigProto &algorithm_config_proto = kv.value(); in PopulateConvMap() local140 const AlgorithmDesc primary(algorithm_config_proto.algorithm()); in PopulateConvMap()142 algorithm_config_proto.has_algorithm_no_scratch() in PopulateConvMap()144 AlgorithmDesc(algorithm_config_proto.algorithm_no_scratch())) in PopulateConvMap()153 entry = AutotuneEntry<Op>(AlgorithmConfig(algorithm_config_proto)); in PopulateConvMap()
940 explicit AlgorithmConfig(const AlgorithmConfigProto& algorithm_config_proto) { in AlgorithmConfig() argument941 const AlgorithmProto& algorithm_proto = algorithm_config_proto.algorithm(); in AlgorithmConfig()943 if (algorithm_config_proto.optional_scratch_size_case() != in AlgorithmConfig()945 scratch_size_ = algorithm_config_proto.scratch_size(); in AlgorithmConfig()947 if (algorithm_config_proto.optional_algorithm_no_scratch_case() != in AlgorithmConfig()950 algorithm_config_proto.algorithm_no_scratch(); in AlgorithmConfig()978 AlgorithmConfigProto algorithm_config_proto; in ToProto() local980 *algorithm_config_proto.mutable_algorithm() = in ToProto()984 *algorithm_config_proto.mutable_algorithm_no_scratch() = in ToProto()988 algorithm_config_proto.set_scratch_size(scratch_size_.value()); in ToProto()[all …]