Home
last modified time | relevance | path

Searched refs:mutable_algorithm (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_conv_algorithm_picker.cc450 *result.mutable_algorithm() = alg.ToProto(); in AutotuneOneConvRunner()
542 *result.mutable_algorithm() = alg.ToProto(); in AutotuneOneConvRunner()
842 *result.mutable_algorithm() = algorithm_proto; in PickBestAlgorithmNoCacheRocm()
892 *result.mutable_algorithm() = alg.ToProto(); in PickBestAlgorithmNoCacheRocm()
952 *backend_config.mutable_algorithm() = best_algo.algorithm(); in RunOnInstruction()
953 backend_config.mutable_algorithm()->mutable_workspace_size()->set_value( in RunOnInstruction()
H A Djitrt_custom_calls.cc775 auto* algo = descriptor.backend_config.mutable_algorithm(); in GetConvDescriptor()
H A Dir_emitter_unnested.cc1039 auto* algorithm = descriptor.backend_config.mutable_algorithm(); in EmitConvolutionThunk()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dconv_ops_gpu.cc77 *result.mutable_algorithm() = desc.ToProto(); in AutotuneConvImpl()
402 *result.mutable_algorithm() = profile_result.algorithm().ToProto(); in AutotuneUnfusedConv()
420 *result.mutable_algorithm() = profile_algorithm.ToProto(); in AutotuneUnfusedConv()
/aosp_15_r20/external/tensorflow/tensorflow/core/util/autotune_maps/
H A Dautotune_serialize.cc69 *kv.mutable_value()->mutable_algorithm() = in ConvMapToProto()
/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.h842 ::stream_executor::dnn::AlgorithmProto* mutable_algorithm(); in Swap()
1588 inline ::stream_executor::dnn::AlgorithmProto* AlgorithmConfigProto::mutable_algorithm() { in mutable_algorithm() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Ddnn.h980 *algorithm_config_proto.mutable_algorithm() = in ToProto()