Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for WithAssignedDevice (0.12 sec)

  1. tensorflow/compiler/jit/clone_constants_for_better_clustering_test.cc

    }
    
    TEST(CloneConstantsForBetterClusteringTest, HostConstantPlacedOnCpu) {
      Scope root = Scope::NewRootScope().ExitOnError();
      Scope on_gpu = root.WithAssignedDevice(kGPU).WithDevice(kGPU);
      Scope on_cpu = root.WithAssignedDevice(kCPU).WithDevice(kCPU);
    
      Output in0 = ops::Placeholder(on_gpu.WithOpName("in0"), DT_FLOAT);
      Output in1 = ops::Placeholder(on_gpu.WithOpName("in1"), DT_FLOAT);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 8.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/xla_activity_listener_test.cc

      }
    
      TestListener* listener() const { return listener_; }
    
     private:
      TestListener* listener_;
    };
    
    GraphDef CreateGraphDef() {
      Scope root = Scope::NewRootScope().ExitOnError().WithAssignedDevice(
          "/job:localhost/replica:0/task:0/device:CPU:0");
      Output a = ops::Placeholder(root.WithOpName("A"), DT_FLOAT);
      for (int i = 0; i < 5; i++) {
        a = ops::MatMul(root.WithOpName(absl::StrCat("matmul_", i)), a, a);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 5.9K bytes
    - Viewed (0)
Back to top