Search Options

Results per page
Sort
Preferred Languages
Advance

Results 91 - 100 of 126 for attrName (0.19 sec)

  1. tensorflow/c/ops.cc

          TF_ShapeInferenceContext* ctx, const char* attr_name, c_type* val, \
          TF_Status* status) {                                               \
        TF_SetStatus(status, TF_OK, "");                                     \
        cc_type v;                                                           \
        auto* cc_ctx = reinterpret_cast<InferenceContext*>(ctx);             \
        Status s = cc_ctx->GetAttr(attr_name, &v);                           \
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jul 28 22:41:35 UTC 2022
    - 10.9K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api_unified_experimental.h

    // `op_name` must outlive `op`.
    void TF_AbstractOpSetOpName(TF_AbstractOp* op, const char* const op_name,
                                TF_Status* s);
    // `attr_name` must outlive `op`.
    void TF_AbstractOpSetAttrType(TF_AbstractOp* op, const char* const attr_name,
                                  TF_DataType value, TF_Status* s);
    
    void TF_DeleteAbstractTensor(TF_AbstractTensor*);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sun Oct 24 11:27:00 UTC 2021
    - 7K bytes
    - Viewed (0)
  3. tensorflow/c/eager/c_api_unified_experimental.cc

      }
      tsl::Set_TF_Status_from_Status(s, tracing_op->SetOpName(op_name));
    }
    
    void TF_AbstractOpSetAttrType(TF_AbstractOp* op, const char* const attr_name,
                                  TF_DataType value, TF_Status* s) {
      Status status =
          unwrap(op)->SetAttrType(attr_name, static_cast<DataType>(value));
      TF_SetStatus(s, static_cast<TF_Code>(status.code()),
                   absl::StatusMessageAsCStr(status));
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 10:15:17 UTC 2024
    - 9K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tfr/passes/decompose.cc

          } else {
            auto attr_name = compose_func.getArgAttrOfType<StringAttr>(
                arg.index(), kAttrArgumentNameAttr);
            auto attribute = op->getAttr(attr_name.getValue());
            if (!attribute) {
              attribute =
                  compose_func.getArgAttr(arg.index(), kAttrArgumentDefaultAttr);
            }
            if (!attribute && attr_name.getValue() == "out_type") {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 14.6K bytes
    - Viewed (0)
  5. tensorflow/c/c_api_experimental.cc

    void TF_AttrBuilderSetType(TF_AttrBuilder* builder, const char* attr_name,
                               TF_DataType value) {
      auto iter = builder->attr_names.insert(attr_name).first;
      builder->Set(*iter, static_cast<tensorflow::DataType>(value));
    }
    
    void TF_AttrBuilderSetTypeList(TF_AttrBuilder* builder, const char* attr_name,
                                   const TF_DataType* values, int num_values) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 29.4K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tfr/passes/raise_to_tf.cc

        if (operand.index() >= call_op.getNumOperands()) {
          auto attr_name = signature.getArgAttrOfType<StringAttr>(
              operand.index(), kAttrArgumentNameAttr);
          auto attr_value =
              signature.getArgAttr(operand.index(), kAttrArgumentDefaultAttr);
          arg_attrs->push_back(
              rewriter.getNamedAttr(attr_name.getValue(), attr_value));
          continue;
        }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 21.8K bytes
    - Viewed (0)
  7. tensorflow/c/kernels_experimental.h

    // array of length at least `max_values` (ideally set to total_size from
    // TF_OpKernelConstruction_GetAttrSize(ctx, attr_name, &list_size,
    // &total_size)).
    TF_CAPI_EXPORT extern void TF_OpKernelConstruction_GetAttrTensorShape(
        TF_OpKernelConstruction* ctx, const char* attr_name, int64_t* dims,
        size_t num_dims, TF_Status* status);
    
    TF_CAPI_EXPORT extern bool TF_IsRefInput(TF_OpKernelContext* ctx, int i,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Aug 07 14:44:39 UTC 2023
    - 9.4K bytes
    - Viewed (0)
  8. tensorflow/cc/framework/cc_op_gen_util.cc

        const bool use_const = entry.second;
        string attr_name = AvoidCPPKeywords(api_def_attr.rename_to());
    
        string attr_comment;
        if (!api_def_attr.description().empty()) {
          // TODO(keveman): Word wrap and indent this, to handle multi-line
          // descriptions.
          strings::StrAppend(&attr_comment, "* ", attr_name, ": ",
                             api_def_attr.description(), "\n");
        }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Feb 26 00:57:05 UTC 2024
    - 25K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc

          }
        }
    
        auto check_i64_attr =
            [&](const std::string& attr_name) -> mlir::LogicalResult {
          if (!backend_config.contains(attr_name)) {
            return op.emitOpError()
                   << "Missing " << attr_name << " attribute in backend_config";
          }
          auto attr = backend_config.getAs<IntegerAttr>(attr_name);
          if (!attr || !attr.getType().isInteger(64)) {
            return op.emitOpError()
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 154.9K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/transforms/fused_kernel_matcher.cc

          auto sizes = mlir::DenseI32ArrayAttr::get(context, {1, 1, 1, 0});
          auto attr_name =
              StringAttr::get(context, mlir::OpTrait::AttrSizedOperandSegments<
                                           void>::getOperandSegmentSizeAttr());
          attrs.push_back(NamedAttribute(attr_name, sizes));
        }
    
        // Insert fused operation right before the BiasAdd operation to guarantee
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 14.9K bytes
    - Viewed (0)
Back to top