Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 16 for _Arg (0.14 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
  6. 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)
  7. 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)
  8. 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)
  9. 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)
  10. 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)
Back to top