Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 21 for mutable_attr (0.17 sec)

  1. tensorflow/compiler/jit/xla_cluster_util.cc

             std::find(node.output_types().begin(), node.output_types().end(),
                       DT_RESOURCE) != node.output_types().end();
    }
    
    void RemoveFromXlaCluster(NodeDef* node_def) {
      node_def->mutable_attr()->erase(kXlaClusterAttr);
    }
    
    void RemoveFromXlaCluster(Node* node) { node->ClearAttr(kXlaClusterAttr); }
    
    namespace {
    typedef xla_config_registry::XlaGlobalJitLevel XlaGlobalJitLevel;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 29 08:39:39 UTC 2024
    - 21.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/device_compiler.h

      NameAttrList name;
      name.set_name(def.op());
      *name.mutable_attr() = def.attr();
      // Remove the "_class" attribute from the attribute set used to create the
      // compilation cache key. This attribute is information for the colocator
      // and causes false uniqueness between nodes.
      name.mutable_attr()->erase("_class");
      return CompileImpl(compile_options, options, name, args, CompileScope::kOp,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 22.1K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/utils/export_utils.cc

      TF_RETURN_IF_ERROR(ConvertAttributes(attr.getAttrs().getValue(),
                                           /*attrs_to_ignore=*/{}, remove_ref_type,
                                           value->mutable_func()->mutable_attr()));
      return absl::OkStatus();
    }
    
    Status ConvertAttribute(const mlir::StringAttr& attr, AttrValue* value) {
      absl::string_view attr_value(attr.getValue().data(), attr.getValue().size());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 19.7K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/compilability_check_util.cc

                                     NodeDef* node_def) {
      const NameAttrList* name_attr;
      TF_RETURN_IF_ERROR(GetNodeAttr(node.attrs(), attr_name, &name_attr));
      node_def->set_op(name_attr->name());
      *(node_def->mutable_attr()) = name_attr->attr();
      return absl::OkStatus();
    }
    
    absl::StatusOr<std::vector<NodeDef>> MakeCallNodesFromAttribute(
        const Node& node, absl::string_view attr_name,
        absl::string_view call_name) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 30.3K bytes
    - Viewed (0)
  5. tensorflow/c/c_api_function.cc

        return;
      }
    
      auto fdef_or = func->record->mutable_fdef();
      if (!fdef_or.ok()) {
        status->status = fdef_or.status();
        return;
      }
    
      (*(fdef_or.value()->mutable_attr()))[string(attr_name)] = attr_value;
    
      status->status = absl::OkStatus();
    }
    
    void TF_FunctionGetAttrValueProto(TF_Function* func, const char* attr_name,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 13.6K bytes
    - Viewed (0)
  6. tensorflow/compiler/jit/build_xla_ops_pass.cc

                              num_resource_inputs),
          std::back_inserter(result->resource_inputs), IncomingEdgeAsOutput);
    
      result->function.set_name(n->type_string());
      *result->function.mutable_attr() = n->def().attr();
      return absl::OkStatus();
    }
    
    Status CopyIncomingControlEdges(Graph* g, Node* from, Node* to) {
      for (const Edge* e : from->in_edges()) {
        if (e->IsControlEdge()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 24.3K bytes
    - Viewed (0)
  7. tensorflow/c/c_api_experimental.cc

      node_def.set_name(op->Name());
      node_def.set_op(op->Name());
      for (int i = 0; i < num_inputs; ++i) {
        node_def.add_input("dummy_input");
      }
      OperationFromInterface(op)->Attrs().FillAttrValueMap(node_def.mutable_attr());
    
      const tensorflow::OpRegistrationData* op_reg_data;
      status->status =
          tensorflow::OpRegistry::Global()->LookUp(node_def.op(), &op_reg_data);
      if (!status->status.ok()) return;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 29.4K bytes
    - Viewed (0)
  8. tensorflow/c/eager/c_api.cc

    // an attribute.
    void AnnotateEagerRuntimeConstructionContext(
        tensorflow::FunctionDef& function_def) {
      tensorflow::AttrValue value;
      SetAttrValue("kEagerRuntime", &value);
      (*function_def.mutable_attr())["_construction_context"] = value;
    }
    
    }  // namespace
    
    extern "C" {
    
    TFE_ContextOptions* TFE_NewContextOptions() { return new TFE_ContextOptions; }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 08:11:23 UTC 2024
    - 44K bytes
    - Viewed (0)
  9. tensorflow/compiler/jit/mark_for_compilation_pass_test.cc

      FunctionDef noinline = compilable;
      noinline.mutable_signature()->set_name("NoInlineFn");
      AddAttr("_noinline", static_cast<bool>(true), noinline.mutable_attr());
    
      FunctionDefLibrary flib;
      *flib.add_function() = compilable;
      *flib.add_function() = uncompilable;
      *flib.add_function() = noinline;
      FunctionLibraryDefinition flib_def(OpRegistry::Global(), flib);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 14 10:11:10 UTC 2024
    - 79.6K bytes
    - Viewed (0)
  10. tensorflow/compiler/jit/encapsulate_subgraphs_pass_test.cc

      // `PerformStaticShapeInferenceBeforeEncapsulation`.
      for (FunctionDef& fdef : *library->mutable_function()) {
        for (NodeDef& node_def : *fdef.mutable_node_def()) {
          node_def.mutable_attr()->erase("_xla_inferred_shapes");
        }
      }
    
      return s;
    }
    
    Status Encapsulate(GraphDef* graphdef, FunctionDefLibrary* library) {
      std::vector<string> encapsulated_functions;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 113.3K bytes
    - Viewed (0)
Back to top