Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 14 for mutable_attr (0.12 sec)

  1. tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc

          }
    
          (*node_def->mutable_attr())["_handle_dtypes"] = handle_dtypes_attr;
          (*node_def->mutable_attr())["_handle_shapes"] = handle_shapes_attr;
        }
      }
    
      TF_RETURN_IF_ERROR(
          SetShapeAttribute("_output_shapes", arg_type, node_def->mutable_attr()));
    
      DataType dtype;
      TF_RETURN_IF_ERROR(ConvertToDataType(arg_type.getElementType(), &dtype));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 35.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc

          }
    
          (*node_def->mutable_attr())["_handle_dtypes"] = handle_dtypes_attr;
          (*node_def->mutable_attr())["_handle_shapes"] = handle_shapes_attr;
        }
      }
    
      TF_RETURN_IF_ERROR(
          SetShapeAttribute("_output_shapes", arg_type, node_def->mutable_attr()));
    
      DataType dtype;
      TF_RETURN_IF_ERROR(ConvertToDataType(arg_type.getElementType(), &dtype));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 23:04:51 UTC 2024
    - 35.2K bytes
    - Viewed (0)
  3. tensorflow/cc/tools/freeze_saved_model.cc

                                   NodeDef* const_node) {
      const_node->set_name(variable_node.name());
      const_node->set_op("Const");
      (*const_node->mutable_attr())["dtype"] = variable_node.attr().at("dtype");
      variable_value.AsProtoTensorContent(
          (*const_node->mutable_attr())["value"].mutable_tensor());
    }
    
    // Converts a ReadVariableOp NodeDef to an Identity NodeDef.
    void ConvertReadVariableOpToIdentity(const NodeDef& node,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 11 08:05:36 UTC 2023
    - 11.9K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/extract_outside_compilation_pass_test.cc

      std::map<string, int> host_compute_core = {{"0", 1}, {"1", 0}};
      std::vector<string> shape_inference_graphs;
      bool has_outside_compilation;
      NameAttrList name_attrs;
      name_attrs.set_name("cluster");
      *name_attrs.mutable_attr() = attrs;
      TF_CHECK_OK(ExtractOutsideCompilationTest(
          "_xla", "_oc", "cluster", name_attrs, "cluster_rewritten", "host_graph",
          host_compute_core, &fld, &shape_inference_graphs,
    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/tensorflow/translate/export_tf_dialect_op.cc

    // executed on TPU or GPU devices, which do not have `Cast` registered as a
    // runtime OpKernel.
    void RemoveIdentityCast(NodeDef* node_def) {
      auto attr = node_def->mutable_attr();
      if (node_def->op() == "Cast" && attr->contains("SrcT") &&
          attr->contains("DstT") &&
          attr->at("SrcT").type() == attr->at("DstT").type() &&
          attr->contains("Truncate") && !attr->at("Truncate").b()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 11.1K bytes
    - Viewed (0)
  6. 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)
  7. 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)
  8. 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)
  9. 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)
  10. 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)
Back to top