Search Options

Results per page
Sort
Preferred Languages
Advance

Results 21 - 24 of 24 for WithOpName (0.14 sec)

  1. tensorflow/cc/framework/scope.h

    /// created within the scope, and WithOpName() changes the suffix which
    /// otherwise defaults to the type of the op.
    ///
    /// Name examples:
    ///
    ///     Scope root = Scope::NewRootScope();
    ///     Scope linear = root.NewSubScope("linear");
    ///     // W will be named "linear/W"
    ///     auto W = Variable(linear.WithOpName("W"),
    ///                       {2, 2}, DT_FLOAT);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 13 09:08:33 UTC 2024
    - 10.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/device_compiler_test.cc

      std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global()));
      Scope scope = Scope::NewRootScope().ExitOnError();
      auto a = ops::_Arg(scope.WithOpName("A"), DT_INT32, 0);
      auto b = ops::_Arg(scope.WithOpName("B"), DT_INT32, 1);
      auto c = ops::Add(scope.WithOpName("C"), a, b);
      auto d = ops::_Retval(scope.WithOpName("D"), c, 0);
      TF_RETURN_IF_ERROR(scope.ToGraph(graph.get()));
      return graph;
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 19.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/force_xla_constants_on_host_pass_test.cc

      Output in = ops::Placeholder(root, DT_FLOAT);
      Output perm = ops::Const(root, {3, 1, 2, 0});
    
      NameAttrList b_name_attr;
      b_name_attr.set_name("TransposeCall");
      ops::PartitionedCall call(root.WithOpName("call"), {in, perm}, {DT_FLOAT},
                                b_name_attr);
      call.output.front().node()->AddAttr(kXlaMustCompileAttr, true);
    
      std::unique_ptr<Graph> graph;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 4.1K bytes
    - Viewed (0)
  4. tensorflow/cc/framework/gradients_test.cc

    }
    
    TEST_F(GradientsTest, UnreachableInput) {
      auto x = Const(scope_test_, {{1.0, 2.0, 3.0}, {4.0, 5.0, 6.0}});
      auto y = Const(scope_test_, {{1.0}, {2.0}, {3.0}});
      auto z = Const(scope_test_.WithOpName("z"), {{9.0, 10.0, 11.0}});
    
      auto m1 = MatMul(scope_test_, x, y);
      auto m2 = MatMul(scope_test_, y, z);
      auto dm1 = Const(scope_test_, {{0.5}, {0.5}});
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 15 15:13:38 UTC 2023
    - 25K bytes
    - Viewed (0)
Back to top