Home
last modified time | relevance | path

Searched refs:set_per_process_gpu_memory_fraction (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dshape_optimizer_test.cc81 ->set_per_process_gpu_memory_fraction(0.1); in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/
H A Dgpu_device_test.cc83 gpu_options->set_per_process_gpu_memory_fraction(
/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.h860 void set_per_process_gpu_memory_fraction(double value); in Swap()
4648 inline void GPUOptions::set_per_process_gpu_memory_fraction(double value) { in set_per_process_gpu_memory_fraction() function