Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 97 for attr_value (0.18 sec)

  1. tensorflow/c/eager/tfe_op_attrs_internal.h

    typedef struct TFE_Op TFE_Op;
    
    namespace tensorflow {
    DEFINE_CONVERSION_FUNCTIONS(tensorflow::AbstractOpAttrs, TFE_OpAttrs);
    
    // Set an AttrValue on the op. Doesn't handle the list types.
    void SetOpAttrValueScalar(TFE_Context* ctx, TFE_Op* op,
                              const tensorflow::AttrValue& default_value,
                              const char* attr_name, TF_Status* status);
    }  // namespace tensorflow
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 10 05:41:19 UTC 2021
    - 1.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_layout_helper.h

      (*op)->setAttr("data_format", StringAttr::get(context, target_data_format));
    
      for (auto pair : shuffle_attrs) {
        StringRef attr_name = pair.first;
        ArrayAttr attr_value = pair.second;
        (*op)->setAttr(attr_name,
                       ShuffleArrayAttr(attr_value, reverse_permutation));
      }
    
      auto fold = cast<FoldOperandsTransposeInterface>(op->getOperation());
      for (unsigned idx : fold.GetLayoutDependentResults()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 08 01:19:25 UTC 2023
    - 5.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_collective.cc

                                         IntegerAttr attr_value, Operation* op,
                                         ModuleOp& module) {
      const auto ex_attr_value = module->getAttrOfType<IntegerAttr>(attr_name);
      if (ex_attr_value == nullptr) {
        module->setAttr(attr_name, attr_value);
        return success();
      }
      if (ex_attr_value == attr_value) {
        return success();
      }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 16K bytes
    - Viewed (0)
  4. tensorflow/c/c_test_util.h

    bool GetFunctionDef(TF_Function* func, tensorflow::FunctionDef* func_def);
    
    bool GetAttrValue(TF_Operation* oper, const char* attr_name,
                      tensorflow::AttrValue* attr_value, TF_Status* s);
    
    // Returns a sorted vector of std::pair<function_name, gradient_func> from
    // graph_def.library().gradient()
    std::vector<std::pair<string, string>> GetGradDefs(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Aug 09 01:06:53 UTC 2018
    - 6K bytes
    - Viewed (0)
  5. tensorflow/compiler/jit/encapsulate_util.cc

    Status AppendToListAttr(Node* n, const string& attr_name, const string& value) {
      std::vector<T> attr_value;
      Status s = GetNodeAttr(n->attrs(), attr_name, &attr_value);
      if (!s.ok() && s.code() != error::NOT_FOUND) {
        return s;
      }
    
      n->ClearAttr(attr_name);
      attr_value.push_back(value);
      n->AddAttr(attr_name, attr_value);
      return absl::OkStatus();
    }
    
    // Replaces attribute value.
    template <typename T>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 15.5K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_with_tf2xla.cc

        }
      }
      return true;
    }
    
    bool HasSymbolRefAttr(Operation* op) {
      for (const auto& attr : op->getAttrs()) {
        Attribute attr_value = attr.getValue();
        if (mlir::isa<SymbolRefAttr>(attr_value)) {
          return true;
        } else if (auto array_attr = mlir::dyn_cast<ArrayAttr>(attr_value)) {
          if (!array_attr.empty() &&
              mlir::isa<SymbolRefAttr>(*array_attr.begin())) {
            return true;
          }
        }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 21:49:50 UTC 2024
    - 9.7K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/translate/export_tf_dialect_op.h

    #include "llvm/ADT/StringRef.h"
    #include "mlir/IR/Operation.h"  // from @llvm-project
    #include "tensorflow/compiler/mlir/tensorflow/utils/export_utils.h"
    #include "tensorflow/core/framework/attr_value.pb.h"
    #include "tensorflow/core/framework/node_def.pb.h"
    #include "tensorflow/core/framework/node_def_util.h"
    #include "tensorflow/core/framework/op_def_builder.h"
    #include "tensorflow/core/platform/status.h"
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 2.4K bytes
    - Viewed (0)
  8. tensorflow/c/c_api_function.cc

                                      const void* proto, size_t proto_len,
                                      TF_Status* status) {
      tensorflow::AttrValue attr_value;
      if (!attr_value.ParseFromArray(proto, proto_len)) {
        status->status = InvalidArgument(
            "Unparseable AttrValue proto passed to "
            "TF_FunctionSetAttrValueProto");
        return;
      }
    
      auto fdef_or = func->record->mutable_fdef();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 13.6K bytes
    - Viewed (0)
  9. tensorflow/c/eager/abstract_op_attrs.h

    #ifndef TENSORFLOW_C_EAGER_ABSTRACT_OP_ATTRS_H_
    #define TENSORFLOW_C_EAGER_ABSTRACT_OP_ATTRS_H_
    
    #include "absl/container/inlined_vector.h"
    #include "tensorflow/core/framework/attr_value.pb.h"
    #include "tensorflow/core/framework/types.pb.h"
    #include "tensorflow/core/platform/status.h"
    
    namespace tensorflow {
    
    // Attributes of an op.
    class AbstractOpAttrs {
     protected:
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 26 22:20:27 UTC 2021
    - 2K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/lite/utils/perception_ops_utils_test.cc

          mlir::NameLoc::get(builder->getStringAttr("fused_func")), "fused_func",
          func_type, {});
    
      func.addEntryBlock();
      mlir::StringAttr attr_value = builder->getStringAttr("MaxUnpooling2D");
      func->setAttr("tf._implements", attr_value);
      return func;
    }
    
    func::FuncOp createMaxUnpoolingFunc(
        mlir::Builder* builder, const SmallVector<int64_t, 4>& input_shape,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Sep 29 21:02:21 UTC 2022
    - 7.4K bytes
    - Viewed (0)
Back to top