Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for _Arg (0.12 sec)

  1. 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)
  2. 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)
  3. 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"
    //
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 21.2K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc

          TF_RET_CHECK(tensor_id.index() == 0)
              << "input port designation not supported";
          // Only assign user of argument the input name if the main graph did not
          // have its _Arg nodes lifted into the functions arguments.
          // Ensure name does not get reused.
          (void)exporter.op_to_name_.GetUniqueName(name);
        }
      }
    
      // Adds nodes for basic block (function) arguments.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 35.2K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc

          TF_RET_CHECK(tensor_id.index() == 0)
              << "input port designation not supported";
          // Only assign user of argument the input name if the main graph did not
          // have its _Arg nodes lifted into the functions arguments.
          // Ensure name does not get reused.
          (void)exporter.op_to_name_.GetUniqueName(name);
        }
      }
    
      // Adds nodes for basic block (function) arguments.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 23:04:51 UTC 2024
    - 35.2K bytes
    - Viewed (0)
Back to top