Search Options

Results per page
Sort
Preferred Languages
Advance

Results 41 - 50 of 125 for attr_name (0.23 sec)

  1. tensorflow/compiler/jit/compilability_check_util.h

                            UncompilableNodesMap* uncompilable_nodes) const;
    
      // Returns compilability of node def retrieved from `node`'s attribute with
      // name `attr_name`.
      bool ExtractNodeDefAndCheckCompilability(
          const Node& node, const std::string& attr_name,
          const std::string& call_name, NameAttrList* encapsulating_function,
          FunctionLibraryRuntime* lib_runtime,
          std::vector<StackFrameView>* stack_trace,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 14.9K bytes
    - Viewed (0)
  2. 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)
  3. 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)
  4. tensorflow/c/c_api_experimental.h

    TF_CAPI_EXPORT extern void TF_AttrBuilderSetType(TF_AttrBuilder* builder,
                                                     const char* attr_name,
                                                     TF_DataType value);
    TF_CAPI_EXPORT extern void TF_AttrBuilderSetTypeList(TF_AttrBuilder* builder,
                                                         const char* attr_name,
                                                         const TF_DataType* values,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 27 21:07:00 UTC 2023
    - 15.1K bytes
    - Viewed (0)
  5. tensorflow/compiler/jit/extract_outside_compilation_pass.cc

        } else if (n->IsIfNode()) {
          for (const string& attr_name :
               std::vector<string>{"then_branch", "else_branch"}) {
            NameAttrList branch_func;
            TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), attr_name, &branch_func));
            (*branch_func.mutable_attr())["_device_ordinal"] = device_ordinal_value;
            n->ClearAttr(attr_name);
            n->AddAttr(attr_name, branch_func);
          }
        } else if (n->IsWhileNode()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 104.7K bytes
    - Viewed (0)
  6. tensorflow/cc/framework/cc_op_gen.cc

          continue;
        }
        const string attr_name =
            api_def_attr.has_default_value()
                ? strings::StrCat("attrs.", api_def_attr.rename_to(), "_")
                : AvoidCPPKeywords(api_def_attr.rename_to());
        strings::StrAppend(&body, spaces, ".Attr(\"", graph_attr.name(), "\", ",
                           attr_name, ")\n");
      }
      strings::StrAppend(&body, "  ;\n");
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 27 17:22:47 UTC 2023
    - 17K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tfr/ir/tfr_ops.td

            all_attrs.insert(attr.getName().strref());
          }
          for (const auto& operand : llvm::enumerate(getArgumentTypes())) {
            if (auto attr_name = getArgAttrOfType<StringAttr>(
                operand.index(), kAttrArgumentNameAttr)) {
              all_attrs.insert(attr_name.getValue());
            }
          }
          return all_attrs;
        }
      }];
    
      let hasVerifier = 1;
      let hasCustomAssemblyFormat = 1;
    }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 10:54:29 UTC 2024
    - 17.4K bytes
    - Viewed (0)
  8. tensorflow/c/c_test_util.cc

      TF_DeleteBuffer(buffer);
      TF_DeleteStatus(s);
      return ret;
    }
    
    bool GetAttrValue(TF_Operation* oper, const char* attr_name,
                      tensorflow::AttrValue* attr_value, TF_Status* s) {
      TF_Buffer* buffer = TF_NewBuffer();
      TF_OperationGetAttrValueProto(oper, attr_name, buffer, s);
      bool ret = TF_GetCode(s) == TF_OK;
      if (ret) ret = attr_value->ParseFromArray(buffer->data, buffer->length);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Oct 15 03:16:52 UTC 2021
    - 17.8K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/translate/import_model.cc

                                 llvm::StringRef attr_name) {
      return attr_name.compare(kOutputShapesAttrName) == 0 &&
             attr_value.value_case() == AttrValue::kList;
    }
    
    bool IsResourceOutputShapesAttribute(const AttrValue& attr_value,
                                         llvm::StringRef attr_name) {
      if (attr_name == "_handle_dtypes" || attr_name == "_handle_shapes")
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 183.2K bytes
    - Viewed (0)
  10. tensorflow/c/kernels_experimental.cc

                                                    const char* attr_name,
                                                    int64_t* dims, size_t num_dims,
                                                    TF_Status* status) {
      ::tensorflow::TensorShape shape;
      auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelConstruction*>(ctx);
      ::tensorflow::Status s = cc_ctx->GetAttr(attr_name, &shape);
      ::tensorflow::Set_TF_Status_from_Status(status, s);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 23 06:12:29 UTC 2024
    - 30.9K bytes
    - Viewed (0)
Back to top