Search Options

Results per page
Sort
Preferred Languages
Advance

Results 41 - 50 of 86 for StringAttr (0.12 sec)

  1. tensorflow/compiler/mlir/quantization/tensorflow/utils/tf_to_xla_attribute_utils.h

                                         int8_t input_zp_value, ArrayAttr strides,
                                         ArrayAttr dilations,
                                         StringAttr conv_padding,
                                         ArrayAttr explicit_paddings,
                                         Value &padding, int num_dims = 4);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sun Dec 10 05:52:02 UTC 2023
    - 2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_device_helper.cc

    }
    
    // Returns true if operation does not have explicit device placement that would
    // prevent it from running on GPU device.
    bool CanUseGpuDevice(Operation *op) {
      auto device_attr = op->getAttrOfType<StringAttr>("device");
      if (!device_attr || device_attr.getValue().empty()) return true;
    
      DeviceNameUtils::ParsedName device;
      if (!DeviceNameUtils::ParseFullName(device_attr.getValue().str(), &device))
        return false;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 21 08:41:18 UTC 2022
    - 2.4K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/utils/nms_utils.cc

                                 StringRef(content.data(), content.size()));
    }
    
    }  // namespace
    
    void ConvertNMSPaddedFunc::RewriteFunc() {
      func_->setAttr(kTFImplements,
                     StringAttr::get(func_.getContext(), kTfNMSPadded));
      Value boxes = func_.getArgument(0);
      Value scores = func_.getArgument(1);
      Value max_output_size = func_.getArgument(2);
      Value iou_threshold = func_.getArgument(3);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.1K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/stablehlo/ops/stablehlo_op_quant_spec.cc

      }
    
      if (GetStableHloQuantConstraints(op)->has_same_scale_requirement) {
        return true;
      }
    
      const bool attr_enforced_quantizable =
          op->hasAttrOfType<StringAttr>(kQuantTraitAttrName) &&
          op->getAttrOfType<StringAttr>(kQuantTraitAttrName).getValue().str() ==
              QuantTraitValues[QuantizationTrait::FullyQuantizable];
      return attr_enforced_quantizable;
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 05:56:10 UTC 2024
    - 7.8K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/cluster_formation.cc

      TF::ReorderOpResultUses(launch_op);
    }
    
    std::string GetDevice(Operation* op) {
      auto device_attr = op->getAttrOfType<StringAttr>("device");
      return device_attr ? device_attr.getValue().str() : "";
    }
    
    bool CanBeIgnoredInCluster(Operation* op) {
      auto device_attr = op->getAttrOfType<StringAttr>("device");
      return !device_attr || device_attr.getValue().empty();
    }
    
    void BuildClusters(Block& block, OpBuilder builder,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Oct 05 13:30:21 UTC 2023
    - 6.8K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tfr/passes/decompose_patterns.td

    include "tensorflow/compiler/mlir/tfr/ir/tfr_ops.td"
    
    class Quantize<string value> : NativeCodeCall<"TFR::Quantize(" # value # ", $0, $1, $_builder)">;
    
    class HasStringAttr<string value> : AttrConstraint<
        CPred<"$_self.cast<StringAttr>().getValue() == \"" # value # "\"">>;
    
    def QuantActRangeNonePattern :
      Pattern<
        (TFR_TFRQuantActRangeOp
         (TFR_ConstOp HasStringAttr<"NONE">:$act), $scale, $zp),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Sep 29 21:02:21 UTC 2022
    - 2.4K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/utils/stablehlo_custom_call.cc

        return failure();
      }
    
      if (auto attr = mlir::dyn_cast<FlatSymbolRefAttr>(f)) {
        return attr;
      }
    
      op.emitOpError() << "'s attribute '" << kCalledFuncAttrName
                       << "' is neither StringAttr nor FlatSymbolRefAttr";
      return failure();
    }
    
    }  // namespace TF
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 2.6K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/transforms/resource_analyzer_test_pass.cc

      StringRef device;
      StringRef container;
      StringRef shared_name;
    };
    
    ResourceKey GetResourceKey(TF::VarHandleOp var_handle_op) {
      ResourceKey resource_key;
    
      if (auto attr = var_handle_op->getAttrOfType<StringAttr>("device")) {
        resource_key.device = attr.getValue();
      }
    
      resource_key.container = var_handle_op.getContainer();
      resource_key.shared_name = var_handle_op.getSharedName();
    
      return resource_key;
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 03 00:21:29 UTC 2023
    - 3K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/quantization/tensorflow/passes/lift_hashtable_ops_as_args.cc

      return true;
    }
    
    // Returns the `shared_name` attribute value if exists. If not, returns an
    // empty string.
    StringRef GetSharedName(Operation* op) {
      if (!op->hasAttrOfType<StringAttr>(kSharedNameAttr)) return "";
      return op->getAttrOfType<StringAttr>(kSharedNameAttr).getValue();
    }
    
    // Checks if the HashTable is initialized. This function assumes that the
    // HashTable is initialized if it appears in the initializer since it can't
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 8.2K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/transforms/graph_pruning.cc

      void runOnOperation() override;
    
     private:
      bool ShouldPreserveOp(Operation* op);
      bool ShouldPreserveIsland(IslandOp island);
      void PruneGraph(GraphOp graph);
    
      llvm::SmallDenseSet<mlir::StringAttr, 4> ops_to_preserve_ids_;
    };
    
    // Checks if a tf_executor.Graph can be pruned.
    // For TensorFlow V1.0 compatibility: when importing a graph without providing
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Oct 05 23:50:19 UTC 2022
    - 7.6K bytes
    - Viewed (0)
Back to top