Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 125 for attr_name (0.55 sec)

  1. tensorflow/c/python_api.cc

    }
    
    void SetAttr(TF_Graph* graph, TF_Operation* op, const char* attr_name,
                 TF_Buffer* attr_value_proto, TF_Status* status) {
      TF_SetAttr(graph, op, attr_name, attr_value_proto, status);
    }
    
    void ClearAttr(TF_Graph* graph, TF_Operation* op, const char* attr_name,
                   TF_Status* status) {
      TF_ClearAttr(graph, op, attr_name, status);
    }
    
    void SetFullType(TF_Graph* graph, TF_Operation* op,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jul 12 18:48:56 UTC 2023
    - 3.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_utils.cc

      return true;
    }
    
    bool GetI32VectorFromDenseI64CompositeAttr(
        const DictionaryAttr& composite_attrs, const std::string& attr_name,
        std::vector<int32_t>* out_vec) {
      DenseIntElementsAttr attr;
      if (!EnsureAttribute<DenseIntElementsAttr>(composite_attrs, attr_name,
                                                 &attr)) {
        return false;
      }
    
      return DenseI64AttrToI32Vector(attr, out_vec);
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 29 18:33:05 UTC 2024
    - 3.4K bytes
    - Viewed (0)
  3. tensorflow/c/kernels.h

    TF_CAPI_EXPORT extern TF_Buffer* TF_OpKernelConstruction_GetAttrFunction(
        TF_OpKernelConstruction* ctx, const char* attr_name, TF_Status* status);
    
    // Return true if the kernel construction has the attr_name
    TF_CAPI_EXPORT extern bool TF_OpKernelConstruction_HasAttr(
        TF_OpKernelConstruction* ctx, const char* attr_name, TF_Status* status);
    
    // Returns the unique operation name for this OpKernel.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jan 09 22:46:22 UTC 2024
    - 24.6K bytes
    - Viewed (0)
  4. tensorflow/c/kernels.cc

      case DataTypeToEnum<type>::value: {                            \
        kernel_builder->cc_builder->TypeConstraint<type>(attr_name); \
        break;                                                       \
      }
    
    void AddTypeConstraint(TF_KernelBuilder* kernel_builder, const char* attr_name,
                           const DataType dtype, TF_Status* status) {
      // This needs to be under tensorflow:: namespace so that
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 22:53:47 UTC 2024
    - 36K bytes
    - Viewed (0)
  5. tensorflow/c/c_api.cc

        }
        desc->node_builder.Attr(attr_name, v);
      }
    }
    
    void TF_SetAttrInt(TF_OperationDescription* desc, const char* attr_name,
                       int64_t value) {
      desc->node_builder.Attr(attr_name, static_cast<int64_t>(value));
    }
    
    void TF_SetAttrIntList(TF_OperationDescription* desc, const char* attr_name,
                           const int64_t* values, int num_values) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 102.3K bytes
    - Viewed (0)
  6. tensorflow/c/eager/c_api.h

    TF_CAPI_EXPORT extern void TFE_OpSetAttrInt(TFE_Op* op, const char* attr_name,
                                                int64_t value);
    TF_CAPI_EXPORT extern void TFE_OpSetAttrFloat(TFE_Op* op, const char* attr_name,
                                                  float value);
    TF_CAPI_EXPORT extern void TFE_OpSetAttrBool(TFE_Op* op, const char* attr_name,
                                                 unsigned char value);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 27 21:07:00 UTC 2023
    - 22.8K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/utils/perception_ops_utils.cc

                                                 const std::string& attr_name,
                                                 int N) {
      ArrayAttr array_attr =
          mlir::dyn_cast_or_null<ArrayAttr>(attrs.get(attr_name));
      if (array_attr == nullptr || array_attr.size() != N) {
        return func->emitWarning()
               << "'" << attr_name << "' attribute for " << kMaxUnpooling
               << " must be set and has size of " << N;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 8.9K bytes
    - Viewed (0)
  8. tensorflow/c/c_api.h

      int64_t total_size;
    } TF_AttrMetadata;
    
    // Returns metadata about the value of the attribute `attr_name` of `oper`.
    TF_CAPI_EXPORT extern TF_AttrMetadata TF_OperationGetAttrMetadata(
        TF_Operation* oper, const char* attr_name, TF_Status* status);
    
    // Fills in `value` with the value of the attribute `attr_name`.  `value` must
    // point to an array of length at least `max_length` (ideally set to
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Oct 26 21:08:15 UTC 2023
    - 82.3K bytes
    - Viewed (0)
  9. tensorflow/compiler/jit/encapsulate_util.cc

      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>
    void ReplaceAttr(Node* n, const string& attr_name, const T& value) {
      n->ClearAttr(attr_name);
      n->AddAttr(attr_name, value);
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 15.5K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_main_function.cc

      for (auto& attr : func->getAttrs()) {
        StringRef attr_name = attr.getName().getValue();
        if (attr_name.starts_with("tf_saved_model.")) {
          func->removeAttr(attr_name);
        }
      }
    
      auto iface = cast<FunctionOpInterface>(func.getOperation());
      for (int i = 0; i < func.getNumArguments(); ++i) {
        for (auto& attr : iface.getArgAttrs(i)) {
          const StringAttr& attr_name = attr.getName();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 16.5K bytes
    - Viewed (0)
Back to top