Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 9 of 9 for StringAttr (1.83 sec)

  1. tensorflow/compiler/mlir/lite/flatbuffer_export.cc

          flex_builder->Bool(name.c_str(),
                             array[i].cast<mlir::BoolAttr>().getValue());
        } else if (llvm::isa<mlir::StringAttr>(attr)) {
          flex_builder->String(name.c_str(),
                               array[i].cast<mlir::StringAttr>().getValue().str());
        } else if (llvm::isa<mlir::vhlo::BooleanV1Attr>(array[i])) {
          flex_builder->Bool(name.c_str(),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:41:49 UTC 2024
    - 164.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc

      // Permute filter sizes operand.
      OpBuilder builder(getOperation());
      auto filter_sizes_permuted = builder.create<TF::DataFormatVecPermuteOp>(
          getLoc(), getFilterSizes(),
          StringAttr::get(getContext(), src_data_format),
          StringAttr::get(getContext(), data_format));
      setOperand(1, filter_sizes_permuted);
    
      return success();
    }
    
    StringRef Conv2DBackpropFilterOp::GetOptimalLayout(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 146.7K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/transforms/optimize.cc

    mlir::Value GetFcOutput(OpBuilder *builder,
                            ::mlir::Operation::result_range result, Value input,
                            Value filter, Value bias,
                            StringAttr fused_activation_function,
                            StringAttr weights_format, BoolAttr keep_num_dims,
                            BoolAttr asymmetric_quantize_inputs) {
      auto fc_op = builder->create<FullyConnectedOp>(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 00:40:15 UTC 2024
    - 102.3K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/translate/import_model.cc

      NameUniquifier* function_name_uniquifier_;
      mlir::StatusScopedDiagnosticHandler error_handler_;
      // All the TF ops encountered that aren't modelled in dialect.
      llvm::DenseSet<mlir::StringAttr> unmodelled_op_names_;
    
     protected:
      // Maps feed as TensorId to new Placeholder node name.
      absl::flat_hash_map<TensorId, absl::string_view> remapped_feeds_;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 183.2K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc

        std::vector<std::string> disabled_checks;
        for (auto attr : op.getDisabledChecks().getAsRange<StringAttr>()) {
          disabled_checks.push_back(attr.getValue().str());
        }
        std::vector<std::string> platforms;
        for (auto attr : op.getPlatforms().getAsRange<StringAttr>()) {
          platforms.push_back(attr.getValue().str());
        }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jun 08 07:28:49 UTC 2024
    - 134.1K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc

        Type result_type = op.getResult(0).getType();
        Value input = op.getOperand(0);
        // Ops with padding is rejected in matching function, so we can use the
        // padding to be 'VALID'.
        StringAttr padding = rewriter.getStringAttr("VALID");
    
        // Use NHWC format.
        int32_t stride_h = 1;
        int32_t stride_w = 1;
        std::optional<ArrayRef<int64_t>> window_strides = op.getWindowStrides();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 09:00:19 UTC 2024
    - 99.8K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/ir/tfl_ops.td

    def TFL_BoolFalse : AttrConstraint<
        CPred<"$_self.isa<BoolAttr>() && !$_self.cast<BoolAttr>().getValue()">,
        "whose value is false">;
    
    class TFL_StringEqualsTo<string value> : AttrConstraint<
        CPred<"$_self.cast<StringAttr>().getValue() == \"" # value # "\"">,
        "whose value equals to '" # value # "'">;
    
    // Ensures the array attribute's size is within the given maximum size.
    class TFL_ArrayMaxCount<int n> : AttrConstraint<
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 19:09:08 UTC 2024
    - 186K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

    // For `NcclAllReduceOp` ops the `device` attribute corresponds to the resource
    // instance.
    std::optional<std::string> NcclAllReduceOp::GetResourceInstanceStr() {
      auto device_attr = (*this)->getAttrOfType<StringAttr>("device");
      // Treat missing device attribute like unspecified (= empty string) attribute.
      // Note that different op instances with the same string (including empty
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 170.8K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/ir/tfl_ops.cc

      }
    }
    
    void buildFusedBroadcastableBinOp(Builder* builder, OperationState& result,
                                      Value lhs, Value rhs,
                                      StringAttr fused_activation_function) {
      auto result_type =
          OpTrait::util::getBroadcastedType(lhs.getType(), rhs.getType());
    
      if (!result_type)
        emitError(result.location)
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 169.2K bytes
    - Viewed (0)
Back to top