Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Ddirect_session_test.cc2405 opts.mutable_fetch_devices()->insert({"y:0", gpu_device_name}); in TEST()
2480 opts.mutable_fetch_devices()->insert({"y:0", gpu_device_name}); in TestFeedAndFetchTensorsInDeviceMemory()
2532 opts.mutable_fetch_devices()->insert({"y:0", gpu_device_name}); in TestFeedAndFetchTensorsInDeviceMemoryFailsToMakeCallable()
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
H A Dgrpc_session_test.cc191 opts.mutable_fetch_devices()->insert({fetch, device_name}); in TEST()
/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.h4150 mutable_fetch_devices(); in Swap()
8451 CallableOptions::mutable_fetch_devices() { in mutable_fetch_devices() function