Search Options

Results per page
Sort
Preferred Languages
Advance

Results 21 - 24 of 24 for withDevices (0.1 sec)

  1. tensorflow/compiler/jit/encapsulate_xla_computations_pass_test.cc

      auto w = ops::Placeholder(scope.WithOpName("W"), DT_RESOURCE);
    
      NameAttrList function;
      function.set_name("launch0");
      auto launch = ops::XlaLaunch(
          scope.WithOpName("launch0").WithDevice("/gpu:0"),
          std::initializer_list<Input>{}, std::initializer_list<Input>{a, b, c, d},
          std::initializer_list<Input>{u, v, w},
          DataTypeVector{DT_FLOAT, DT_INT32, DT_FLOAT, DT_FLOAT}, function);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 16 18:03:15 UTC 2023
    - 14.7K bytes
    - Viewed (0)
  2. tensorflow/cc/framework/scope.cc

      return Scope(new Impl(*this, Impl::Tags::ControlDeps(),
                            std::vector<Operation>(),
                            /* clear_control_deps */ true));
    }
    
    Scope Scope::WithDevice(const string& device) const {
      return Scope(new Impl(*this, Impl::Tags::Device(), device));
    }
    
    Scope Scope::WithAssignedDevice(const string& assigned_device) const {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 13 05:57:22 UTC 2024
    - 20.9K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/encapsulate_subgraphs_pass_test.cc

      }
      std::sort(edges.begin(), edges.end());
      return edges;
    }
    
    TEST(EncapsulateSubgraphsTest, InputDeduplication) {
      Scope root = Scope::NewRootScope().ExitOnError().WithDevice(
          "/job:localhost/replica:0/task:0/cpu:0");
      auto x = ops::Placeholder(root.WithOpName("x"), DT_FLOAT);
      auto add1 = ops::Add(root.WithOpName("add1"), x, x);
      add1.node()->AddAttr("_cluster", "cluster1");
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 113.3K bytes
    - Viewed (0)
  4. tensorflow/cc/gradients/math_grad.cc

      // [[[105, 192], [105, 192], [105, 192]], [[0, 180], [0, 180], [0, 180]]]
      auto grad_tiled = Tile(scope, grad, tile_scaling);
    
      Scope cpu_scope = scope.WithDevice("/cpu:0");
    
      // [3]
      auto rank = Rank(cpu_scope, op.input(0));
    
      // Normalize any negative indices in the reduction_axes to positive values.
      auto reduction_indices_pos =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Aug 25 18:20:20 UTC 2023
    - 50.7K bytes
    - Viewed (0)
Back to top