Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/
H A Dgpu_device_test.cc81 GPUOptions* gpu_options = config->mutable_gpu_options();
473 opts.config.mutable_gpu_options() in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dshape_optimizer_test.cc80 session_options.config.mutable_gpu_options() in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_graph_test.cc80 GPUOptions* gpu_options = config->mutable_gpu_options(); in TEST()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dcollective_test_util.cc122 auto* virtual_devices = options.config.mutable_gpu_options() in CreateGPUDevices()
H A Dcollective_param_resolver_local_test.cc67 config.mutable_gpu_options() in RunCompleteDefaultRanking()
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dc_api_test_util.cc473 server_def.mutable_default_session_config()->mutable_gpu_options(); in GetMultiClientServerDef()
/aosp_15_r20/external/tensorflow/tensorflow/c/
H A Dc_api_experimental.cc149 auto* gpu_options = config.mutable_gpu_options(); in TF_CreateConfig()
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/protobuf/
Dconfig.pb.h2546 ::tensorflow::GPUOptions* mutable_gpu_options(); in Swap()
6537 inline ::tensorflow::GPUOptions* ConfigProto::mutable_gpu_options() { in mutable_gpu_options() function