Searched refs:WithAssignedDevice (Results 1 – 7 of 7) sorted by relevance
65 Scope on_gpu = root.WithAssignedDevice(kGPU).WithDevice(kGPU); in TEST()66 Scope on_cpu = root.WithAssignedDevice(kCPU).WithDevice(kCPU); in TEST()92 Scope on_gpu = root.WithAssignedDevice(kGPU).WithDevice(kGPU); in TEST()118 Scope on_gpu = root.WithAssignedDevice(kGPU).WithDevice(kGPU); in TEST()119 Scope on_cpu = root.WithAssignedDevice(kCPU).WithDevice(kCPU); in TEST()147 Scope on_gpu = root.WithAssignedDevice(kGPU).WithDevice(kGPU); in TEST()148 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 …]
248 Scope host_scope = main_scope.WithAssignedDevice(host_name); in ConvertTensorFlowSliceToStaticShapedSlice()267 ops::Slice(main_scope.WithAssignedDevice(slice->assigned_device_name()) in ConvertTensorFlowSliceToStaticShapedSlice()
330 .WithAssignedDevice(device); in ReplaceNodeWithXlaCompileAndXlaRun()
144 Scope WithAssignedDevice(const string& assigned_device) const;
453 Scope Scope::WithAssignedDevice(const string& assigned_device) const { in WithAssignedDevice() function in tensorflow::Scope
637 Scope s = Scope::NewRootScope().ExitOnError().WithAssignedDevice("/gpu:0"); in TEST_F()