- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 111 for _retval (0.15 sec)
-
tensorflow/compiler/jit/encapsulate_xla_computations_pass_test.cc
add_attrs(g.node()); auto out0 = ops::_Retval(scope.WithOpName("b_identity_0_retval_RetVal"), b_identity, 0); auto out1 = ops::_Retval(scope.WithOpName("e_0_retval_RetVal"), e, 1); auto out2 = ops::_Retval(scope.WithOpName("g_0_retval_RetVal"), g, 2); auto out3 = ops::_Retval(scope.WithOpName("readu_0_retval_RetVal"), read_u, 3);
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/tensorflow/tests/mlir2graphdef/preserve-entry-func-names.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 17 22:54:55 UTC 2023 - 1.3K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_xla_computations_pass.cc
// Computes the permutation to produce the correct retval order, and update // the argument indices. for (int i = 0; i < num_retvals; ++i) { int index; TF_RETURN_IF_ERROR(GetIndexAttr(*retvals[i], num_retvals, &index)); (*output_permutation)[index] = i; retvals[i]->AddAttr("index", i); } AddNodeAttr(kXlaClusterIdAttr, call_def->name(), call_def);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 15.1K bytes - Viewed (0) -
tensorflow/compiler/jit/extract_outside_compilation_pass.cc
// Builds XlaSendFromHost node, and replaces all _Retval nodes with it. absl::StatusOr<Node*> ReplaceRetNodesWithSendFromHostNode( Graph* g, const string& oc_cluster_name, std::vector<DataType>* send_from_host_dtypes, Node* key_placeholder) { // TODO(b/77601805): use in nodes for sink node, instead of traversing all // nodes. std::vector<Node*> ret_nodes = GatherNodesWithType(*g, "_Retval");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 104.7K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_compile_util_test.cc
EXPECT_EQ(arg_node->op_def().name(), "_Arg"); EXPECT_EQ(arg_node->attrs().FindByString("T")->type(), DT_FLOAT); const Node* retval_node = node_name_index.at("_retval0"); EXPECT_EQ(retval_node->op_def().name(), "_Retval"); EXPECT_EQ(retval_node->attrs().FindByString("T")->type(), DT_FLOAT); EXPECT_EQ(identity_node->num_outputs(), 1); EXPECT_EQ(retval_node->num_inputs(), 1);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 16 21:48:05 UTC 2023 - 6K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_subgraphs_pass.cc
const Edge* edge, const absl::flat_hash_map<const Node*, Node*>& node_images); // Creates a _Retval node for the src node of edge, and add it to results_, // if none exists yet. If a new _Retval node is created, also adds the edge // within the subgraph from the src to the _Retval node. Status RecordResult( const Edge* edge, const absl::flat_hash_map<const Node*, Node*>& node_images);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 51K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/parse_example_v2.mlir
// CHECK-NEXT: } tf_executor.fetch %outputs_10#0 : tensor<?x2xi64> } func.return %0#0 : tensor<?x2xi64> // CHECK: name: "ParseExample/ParseExampleV2" // CHECK-NEXT: op: "_Retval" // CHECK-NEXT: input: "ParseExample" }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Aug 14 15:35:49 UTC 2023 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_activity_listener_test.cc
count: 4 } } unclustered_op_histogram { op: "NoOp" count: 2 } unclustered_op_histogram { op: "_Arg" count: 1 } unclustered_op_histogram { op: "_Retval" count: 1 } } )", &expected_auto_clustering_activity); EXPECT_EQ(listener()->auto_clustering_activity().DebugString(), expected_auto_clustering_activity.DebugString());
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/compiler/jit/pjrt_compile_util_test.cc
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; } StatusOr<FunctionDef> SampleFuntionAddXY(const std::string& name) { TF_ASSIGN_OR_RETURN(auto graph, SampleGraphAddXY());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Aug 21 23:21:57 UTC 2023 - 4.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/saved_model_export_test.cc
StrEq("input_tensor:0")); // Match the `_Retval` node that corresponds to the return value of @main. const auto retval_node_itr = llvm::find_if(exported_model->graph_def().node(), [](const NodeDef& node) { return node.op() == "_Retval"; }); ASSERT_NE(retval_node_itr, exported_model->graph_def().node().end());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 20 11:11:25 UTC 2024 - 19.6K bytes - Viewed (0)