Searched refs:WithAssignedDevice (Results 1 – 10 of 10) sorted by relevance
66 Scope on_gpu = root.WithAssignedDevice(kGPU).WithDevice(kGPU); in TEST()67 Scope on_cpu = root.WithAssignedDevice(kCPU).WithDevice(kCPU); in TEST()93 Scope on_gpu = root.WithAssignedDevice(kGPU).WithDevice(kGPU); in TEST()119 Scope on_gpu = root.WithAssignedDevice(kGPU).WithDevice(kGPU); in TEST()151 Scope on_gpu = root.WithAssignedDevice(kGPU).WithDevice(kGPU); in TEST()152 Scope on_cpu = root.WithAssignedDevice(kCPU).WithDevice(kCPU); in TEST()180 Scope on_gpu = root.WithAssignedDevice(kGPU).WithDevice(kGPU); in TEST()181 Scope on_cpu = root.WithAssignedDevice(kCPU).WithDevice(kCPU); in TEST()
103 .WithAssignedDevice(kDeviceName) in TEST()149 .WithAssignedDevice(kDeviceName) in TEST()169 .WithAssignedDevice(kDeviceName) in TEST()195 .WithAssignedDevice(kDeviceName) in TEST()213 .WithAssignedDevice(kDeviceName) in TEST()242 Scope::NewRootScope().ExitOnError().WithAssignedDevice(kDeviceName); in TEST()260 .WithAssignedDevice(kDeviceName) in TEST()279 .WithAssignedDevice(kDeviceName) in TEST()301 .WithAssignedDevice(kDeviceName) in TEST()331 .WithAssignedDevice(kDeviceName) in TEST()[all …]
103 Scope root = Scope::NewRootScope().WithAssignedDevice(kGPU0).ExitOnError(); in TEST()115 Scope root = Scope::NewRootScope().WithAssignedDevice(kCPU0).ExitOnError(); in TEST()127 Scope root = Scope::NewRootScope().WithAssignedDevice(kGPU0).ExitOnError(); in TEST()130 Output b = ops::Add(root.WithOpName("b").WithAssignedDevice(kGPU1), a, a); in TEST()
138 .WithAssignedDevice(cpu_device), in MergeOutgoingDataEdges()153 .WithAssignedDevice(new_node->assigned_device_name()), in MergeOutgoingDataEdges()489 .WithAssignedDevice(device_name_str); in ReplaceNodeWithXlaCompileAndXlaRun()
248 Scope host_scope = main_scope.WithAssignedDevice(host_name); in ConvertTensorFlowSliceToStaticShapedSlice()267 ops::Slice(main_scope.WithAssignedDevice(slice->assigned_device_name()) in ConvertTensorFlowSliceToStaticShapedSlice()
79 Scope root = Scope::NewRootScope().ExitOnError().WithAssignedDevice( in CreateGraphDef()
279 .WithAssignedDevice(kXlaDeviceName) in TEST_F()
144 Scope WithAssignedDevice(const string& assigned_device) const;
453 Scope Scope::WithAssignedDevice(const string& assigned_device) const { in WithAssignedDevice() function in tensorflow::Scope
644 Scope s = Scope::NewRootScope().ExitOnError().WithAssignedDevice("/gpu:0"); in TEST_F()