Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 87 for _retval (0.12 sec)

  1. tensorflow/compiler/jit/rearrange_function_argument_pass_test.cc

        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);
        auto ret2 = ops::_Retval(s.WithOpName("ret2"), arg2, 2);
        std::unique_ptr<Graph> g(new Graph(OpRegistry::Global()));
        TF_CHECK_OK(s.ToGraph(g.get()));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Feb 09 11:36:41 UTC 2024
    - 10.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/extract_outside_compilation_pass_test.cc

      {
        tensorflow::Scope s = tensorflow::Scope::NewRootScope();
        Output arg = ops::_Arg(s.WithOpName("arg"), DT_INT32, 0);
        Output identity = ops::Identity(s.WithOpName("identity_true_fn"), arg);
        ops::_Retval retval(s.WithOpName("retval"), identity, 0);
        std::unique_ptr<Graph> g(new Graph(OpRegistry::Global()));
        TF_CHECK_OK(s.ToGraph(g.get()));
        auto node_name_image = g->BuildNodeNameIndex();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 41K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/tests/legalize_tfg.mlir

        // CHECK: tf_executor.island wraps "tf._Retval"(%outputs_0) {T = !tf_type.resource, _mlir_name = "func_call", index = 0 : i64} : (tensor<*x!tf_type.resource>) -> ()
        %ctl_1 = _Retval(%test_func_name0) name("func_call") {T = !tf_type.resource, index = 0 : i64} : tensor<*x!tf_type.resource>
        // CHECK: tf_executor.fetch
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 30 06:52:55 UTC 2023
    - 2.9K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/xla_compile_util.cc

        TF_RETURN_IF_ERROR(status);
        graph->AddEdge(node, 0, main_node, i);
      }
    
      // Similarly with return values, create dummy _Retval nodes fed by `node`.
      for (int64_t i = 0, end = result_types.size(); i < end; ++i) {
        Node* node;
        string retval_name = absl::StrCat("_retval", i);
        Status status = NodeBuilder(retval_name, FunctionLibraryDefinition::kRetOp)
                            .Input(main_node, i)
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 09:53:30 UTC 2024
    - 4.6K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/quantization/tensorflow/python/save_model.py

      might become the name of Retval nodes as well (with an index suffix if there
      are multiple output tensors from one node). Since Retval nodes are not used in
      SavedModel, this function removes them and restore the names to the actual
      output tensors.
    
      Args:
        graph_def: the converted GraphDef.
    
      Returns:
        The GraphDef with Retval nodes removed and output tensor names restored.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 01:09:50 UTC 2024
    - 12.3K bytes
    - Viewed (0)
  6. 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)
  7. 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)
  8. 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)
  9. 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)
  10. 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)
Back to top