- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 16 for _Arg (0.07 sec)
-
tensorflow/compiler/jit/rearrange_function_argument_pass_test.cc
// "ret0" = "arg1" // "ret1" = "arg0" tensorflow::Scope s = tensorflow::Scope::NewRootScope(); Output arg0 = ops::_Arg(s.WithOpName("arg0"), DT_RESOURCE, 0); Output arg1 = ops::_Arg(s.WithOpName("arg1"), DT_RESOURCE, 1); Output arg2 = ops::_Arg(s.WithOpName("arg2"), DT_INT32, 2); auto ret0 = ops::_Retval(s.WithOpName("ret0"), arg1, 0); auto ret1 = ops::_Retval(s.WithOpName("ret1"), arg0, 1);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 11:36:41 UTC 2024 - 10.5K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_xla_computations_pass_test.cc
auto arg0 = ops::_Arg(scope.WithOpName("a_0_arg"), DT_INT32, 0); auto arg1 = ops::_Arg(scope.WithOpName("b_0_arg"), DT_FLOAT, 1); auto arg2 = ops::_Arg(scope.WithOpName("c_0_arg"), DT_INT32, 2); auto arg3 = ops::_Arg(scope.WithOpName("d_0_arg"), DT_FLOAT, 3); auto arg4 = ops::_Arg(scope.WithOpName("u_0_arg"), DT_RESOURCE, 4); auto arg5 = ops::_Arg(scope.WithOpName("v_0_arg"), DT_RESOURCE, 5);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 16 18:03:15 UTC 2023 - 14.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/saved_model_export_test.cc
} TEST(CreateSaverDefTest, CreateValidSaverDef) { // Needs to have a _Arg node with an attribute "tf_saved_model.index_path" = // ["__tf_file_prefix"]. GraphDef graph_def; ASSERT_TRUE(TextFormat::ParseFromString( R"pb(node { name: "foo", op: "_Arg", attr { key: "tf_saved_model.index_path",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 20 11:11:25 UTC 2024 - 19.6K bytes - Viewed (0) -
tensorflow/compiler/jit/extract_outside_compilation_pass_test.cc
// "add" = "arg0" + "arg1" // "ret0" = "add" // "ret1" = "arg1" tensorflow::Scope s = tensorflow::Scope::NewRootScope(); Output arg0 = ops::_Arg(s.WithOpName("arg0"), DT_INT32, 0); Output arg1 = ops::_Arg(s.WithOpName("arg1"), DT_FLOAT, 1); Output arg2 = ops::_Arg(s.WithOpName("arg2"), DT_INT32, 2); Output add = ops::Add(s.WithOpName("add"), arg0, arg0); auto ret0 = ops::_Retval(s.WithOpName("ret0"), add, 0);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 41K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/mlir_bridge_pass_util_test.cc
graph.SetConstructionContext(ConstructionContext::kEagerRuntime); tensorflow::set_tf2_execution(true); ConfigProto config = ConfigProto(); Scope root = Scope::NewRootScope().ExitOnError(); Output a = ops::_Arg(root.WithOpName("A"), DT_RESOURCE, 0); std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(a.node())}); Node* call; NameAttrList f_name_attr; f_name_attr.set_name(fd.signature().name()); TF_ASSERT_OK(
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/shape_inference.cc
<< context->DebugString(handle); } } int index = -1; if (n->type_string() == "_Arg") { // NOTE: during runtime, Placeholder ops will be replaced as `_Arg` ops. // And Args must have `index` attribute. TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), "index", &index)); } else if (n->type_string() == "Placeholder") {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 00:41:19 UTC 2024 - 13K bytes - Viewed (0) -
tensorflow/compiler/jit/compilability_check_util.cc
LogNotCompilable(node, uncompilable_reason); return false; } // _Arg nodes in a top-level function represent feeds and _Retval nodes in a // top-level function represent fetches. if (stack_depth == 1 && (node.type_string() == "_Arg" || node.type_string() == "_Retval")) { absl::string_view uncompilable_reason = "top level _Arg or _Retval"; MaybeMarkUncompilableNode(uncompilable_reason, *stack_trace,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 30.3K bytes - Viewed (0) -
tensorflow/compiler/jit/device_compiler_test.cc
} absl::StatusOr<std::unique_ptr<Graph>> SampleGraphAddXY() { 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()));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 19.8K bytes - Viewed (0) -
tensorflow/compiler/jit/compilability_check_util.h
// xla::ComputationBuilder assigns. How does this handle gets assigned for // constant arguments? Even constant arguments get an _Arg node in the graph // instantiated for Function compilation. The tf2xla kernel for constant _Arg // nodes takes the constant value, converts it to XlaLiteral, and feeds it // to xla::ComputationBuilder.ConstantLiteral, which returns the handle. This
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 14.9K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_xla_computations_pass.cc
if (!TryGetNodeAttr(n.attrs(), "_is_guaranteed_constant", &guaranteed_constant)) { return false; } return guaranteed_constant; } // Finds the `index` of an _Arg or _Retval node. Status GetIndexAttr(const Node& n, int num_args, int* index) { TF_RETURN_IF_ERROR(GetNodeAttr(n.attrs(), "index", index)); if (*index < 0 || *index >= num_args) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 15.1K bytes - Viewed (0)