Searched refs:mutable_fetch_devices (Results 1 – 3 of 3) sorted by relevance
2405 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()
191 opts.mutable_fetch_devices()->insert({fetch, device_name}); in TEST()
4150 mutable_fetch_devices(); in Swap()8451 CallableOptions::mutable_fetch_devices() { in mutable_fetch_devices() function