- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 32 for WithOpName (0.17 sec)
-
tensorflow/compiler/jit/build_xla_ops_pass.cc
// outputs = Merge(tf_call_outputs, xla_run_outputs). ops::Switch s(root.WithOpName("predicated_compilation_key"), xla_compile.key, xla_compile.compilation_successful); Output predicated_compilation_key = s.output_true; Output inverse_predicated_compilation_key = s.output_false; ops::_XlaRun xla_run(root.WithOpName("xla_run"), xla_run_args,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 24.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/mlir_bridge_pass_util_test.cc
Scope root = Scope::NewRootScope().ExitOnError(); Output x = ops::Placeholder(root.WithOpName("x"), DT_FLOAT); NameAttrList f_name_attr; f_name_attr.set_name("XTimesTwoFloat"); ops::TPUPartitionedCall f(root.WithOpName("f"), {x}, /*device_ordinal=*/0, {DT_FLOAT}, f_name_attr); TF_ASSERT_OK(root.ToGraph(&graph));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 19:51:50 UTC 2024 - 10.3K bytes - Viewed (0) -
tensorflow/compiler/jit/compilability_check_util_test.cc
Scope root = Scope::NewRootScope().ExitOnError(); TF_ASSERT_OK(root.graph()->AddFunctionLibrary(flib)); auto predicate = ops::Placeholder(root.WithOpName("pred"), DT_BOOL); auto placeholder = ops::Placeholder(root.WithOpName("A"), DT_INT32); std::vector<NodeBuilder::NodeOut> if_inputs( {NodeBuilder::NodeOut(placeholder.node())}); Node* if_node; TF_ASSERT_OK(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jun 10 12:32:39 UTC 2022 - 22.3K bytes - Viewed (0) -
tensorflow/compiler/jit/build_xla_ops_pass_test.cc
} Node* MakeWrite(const Scope& scope, Output value_to_write, const string& id) { Output var_handle = ops::VarHandleOp(scope.WithOpName("Var_" + id), DT_FLOAT, TensorShape({})); ops::AssignVariableOp assign_op(scope.WithOpName("Assignee_" + id), var_handle, value_to_write); return assign_op.operation.node(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 12.2K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_activity_listener_test.cc
"/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); a = ops::Add(root.WithOpName(absl::StrCat("add_", i)), a, a); } GraphDef graph_def; root.graph()->ToGraphDef(&graph_def); return graph_def; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 5.9K bytes - Viewed (0) -
tensorflow/cc/framework/fuzzing/cc_op_fuzz_gen.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jan 27 16:26:51 UTC 2024 - 13K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_subgraphs_pass_test.cc
auto x1 = ops::Placeholder(root.WithOpName("x1"), DT_FLOAT); auto x2 = ops::Placeholder(root.WithOpName("x2"), DT_FLOAT); auto const_guarantee_x2 = ops::GuaranteeConst(root.WithOpName("const_guarantee_x2"), x2); auto const_guarantee_x1 = ops::GuaranteeConst(root.WithOpName("const_guarantee_x1"), x1); auto add1 = ops::Add(root.WithOpName("add1"), const_guarantee_x1, const_guarantee_x2);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 113.3K bytes - Viewed (0) -
tensorflow/compiler/jit/device_executable_persistor_test.cc
auto a = ops::_Arg(scope.WithOpName("A"), DT_INT32, 0); auto b = ops::_Arg(scope.WithOpName("B"), DT_INT32, 1); if (mul) { auto c = ops::Mul(scope.WithOpName("C"), a, b); auto d = ops::_Retval(scope.WithOpName("D"), c, 0); TF_RETURN_IF_ERROR(scope.ToGraph(graph.get())); } else { auto c = ops::Add(scope.WithOpName("C"), a, b);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 25.9K bytes - Viewed (0) -
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) -
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)