- Sort Score
- Result 10 results
- Languages All
Results 31 - 32 of 32 for WithOpName (0.1 sec)
-
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) -
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)