- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 127 for _retval (0.48 sec)
-
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) -
tensorflow/compiler/jit/extract_outside_compilation_pass.h
// following steps: // // 1. Add a XLA computation key placeholder node (it will be used as input for // XlaRecvAtHost and XlaSendFromHost); // 2. Replace all _Arg nodes with one single XlaRecvAtHost node; // 3. Replace all _Retval nodes with one single XlaSendFromHost node; // 4. Mark all nodes except key placeholder with attr `xla_cluster_attr_name` // and `outside_compilation_attr_name`;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 5.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/utils/utils.cc
std::string GetComposeFuncName(StringRef tf_op_name) { std::string compose_func_name; for (int i = 0; i < tf_op_name.size(); ++i) { if (tf_op_name[i] == '_') { // The field name must not contain "_"s. "_Arg" and "_RetVal" are special // op names and we can return empty string to skip the decomposition. return {}; } if (tf_op_name[i] == '.') { compose_func_name.push_back('_');
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 06 03:08:33 UTC 2023 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/jit/compilability_check_util.cc
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_executable_persistor_test.cc
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); 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 - 25.9K bytes - Viewed (0) -
tensorflow/compiler/jit/tests/opens2s_gnmt_mixed_precision.golden_summary
Sum 2 Switch 247 TensorArrayGatherV3 8 TensorArrayGradV3 17 TensorArrayReadV3 9 TensorArrayScatterV3 8 TensorArraySizeV3 4 TensorArrayV3 8 TensorArrayWriteV3 9 Unique 2 VariableV2 164 _Retval 5 cluster 0 size 440 Abs 40 AddN 1 Any 41 Cast 40 ConcatV2 2 Const 95 ExpandDims 2 IsInf 1 IsNan 40 L2Loss 40 LogicalOr 1 Max 41 Minimum 1 Mul 82 Pack 3
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jan 06 10:38:14 UTC 2023 - 5K bytes - Viewed (0) -
tensorflow/compiler/jit/device_compiler_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; } absl::StatusOr<FunctionDef> SampleFuntionAddXY(const std::string& name) {
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/mlir/tensorflow/transforms/tfg-to-tfe.cc
static const char kImportModelDefaultGraphFuncName[] = "main"; // Please refer to the TFG dialect description for the list of used attributes. // Belows are the attributes in TFE. // TFE Arguments and Results (Got from "_Arg", // "_Retval", .etc) // NodeDef.device <-> "tf.device" // NodeDef.attr <-> "tf." // // TFE general operations // NodeDef.device <-> "device" // // The following two functions are only used for mapping/excluding attributes
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.2K bytes - Viewed (0) -
tensorflow/compiler/jit/mark_for_compilation_pass_test.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 14 10:11:10 UTC 2024 - 79.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
mlir::OperationName name = inner_op->getName(); if (!name.isRegistered() && // Skip unmodelled ops that are handled differently. (node_type_name != "_Arg" && node_type_name != "_Retval") && !unmodelled_op_names_.count(name.getIdentifier())) { if (node.op_def().is_stateful()) { VLOG(1) << "[potentially conservative] Op type `" << node.type_string()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0)