Home
last modified time | relevance | path

Searched refs:use_cuda_malloc_async (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.-g-p-u-options.pbtxt118 name: "use_cuda_malloc_async"
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/
H A Dgpu_device_test.cc77 const bool use_cuda_malloc_async = false) { argument
86 use_cuda_malloc_async);
H A Dgpu_process_state.cc221 options.experimental().use_cuda_malloc_async()) { in GetGPUAllocator()
/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/
H A Dconfig.proto235 bool use_cuda_malloc_async = 11; field
/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.h623 bool use_cuda_malloc_async() const; in Swap()
4596 inline bool GPUOptions_Experimental::use_cuda_malloc_async() const { in use_cuda_malloc_async() function