Search Options

Results per page
Sort
Preferred Languages
Advance

Results 31 - 32 of 32 for WithOpName (0.1 sec)

  1. 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)
  2. 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