Search Options

Results per page
Sort
Preferred Languages
Advance

Results 51 - 60 of 94 for ret_attrs (0.19 sec)

  1. tensorflow/compiler/mlir/lite/experimental/tac/transforms/device_transform.cc

          }
        }
    
        // Build the new float-versioned op.
        OperationState state(op->getLoc(), op->getName());
        state.operands = dequantized_inputs;
        state.types = result_types;
        state.attributes = op->getAttrs();
        state.successors = op->getSuccessors();
        builder->setInsertionPoint(op);
        Operation* new_op = builder->create(state);
    
        // Insert quantize ops for every outputs and rewrite.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize.cc

          OperationState new_state(quantizing_op->getLoc(),
                                   quantizing_op->getName().getStringRef(), inputs,
                                   output_types, quantizing_op->getAttrs());
          for (int i = 0; i < quantizing_op->getNumRegions(); ++i) {
            new_state.addRegion();
          }
          Operation* quantized_op = rewriter.create(new_state);
          if (quantizing_op->getNumRegions() != 0) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 22 05:52:39 UTC 2024
    - 23.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting_cleanup.cc

      }
    
      // Rebuild the new operation with lesser number of results.
      OpBuilder builder(op);
      Operation *new_op = Operation::create(
          op->getLoc(), op->getName(), new_result_types, op->getOperands(),
          op->getAttrs(), op->getPropertiesStorage(), op->getSuccessors(),
          op->getNumRegions());
      builder.insert(new_op);
    
      // Move region bodies to the new operation.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 19.7K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/stablehlo/passes/convert_func_to_bfloat16.cc

        // OperationState so we can add regions to the new op.
        OperationState state(op->getLoc(), op->getName().getStringRef(), operands,
                             new_results, op->getAttrs(), op->getSuccessors());
        for (Region& region : op->getRegions()) {
          Region& new_region = *state.addRegion();
          rewriter.inlineRegionBefore(region, new_region, new_region.begin());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 9.3K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/xla_call_module_serialization.cc

      auto original_stablehlo_module_attrs =
          op->getAttrOfType<DictionaryAttr>(kStablehloModuleAttrsAttrName);
      if (original_stablehlo_module_attrs) {
        (*stablehlo_module)->setAttrs(original_stablehlo_module_attrs);
        // Now, remove the attribute because later passes may not know how to handle
        // it, we may encounter errors such as:
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 9.5K bytes
    - Viewed (0)
  6. tensorflow/compiler/jit/node_matchers.h

        input_matchers_ = std::move(inputs);
      }
    
      void set_control_deps(NodeSeqMatcher control_deps) {
        DCHECK(IsEmpty());
        control_deps_ = std::move(control_deps);
      }
    
      void set_attr(AttrKeyValuePair attr) {
        DCHECK(IsEmpty());
        attr_ = std::move(attr);
      }
    
      bool IsEmpty() const {
        return !name().has_value() && !op().has_value() && !inputs().has_value() &&
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 8.1K bytes
    - Viewed (0)
  7. staging/src/k8s.io/apiserver/pkg/storage/selection_predicate_test.go

    			panic(err)
    		}
    		parsedField, err := fields.ParseSelector(item.fieldSelector)
    		if err != nil {
    			panic(err)
    		}
    		sp := &SelectionPredicate{
    			Label: parsedLabel,
    			Field: parsedField,
    			GetAttrs: func(runtime.Object) (label labels.Set, field fields.Set, err error) {
    				return item.labels, item.fields, item.err
    			},
    		}
    		got, err := sp.Matches(&Ignored{})
    		if e, a := item.err, err; e != a {
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Thu Nov 30 10:39:09 UTC 2023
    - 8.6K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/utils/attribute_utils.h

    // Copies attributes that satisfy the given predicate from `from` to `to`.
    template <typename Predicate>
    void CopyAttributes(Operation *from, Operation *to, Predicate P) {
      for (const NamedAttribute &attr : from->getAttrs())
        if (P(attr)) to->setAttr(attr.getName(), attr.getValue());
    }
    
    // Copies attributes whose name begins with an _ from `from` to `to`.
    inline void CopyUnderscoredAttributes(Operation *from, Operation *to) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 22 19:47:48 UTC 2024
    - 8.7K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/quantization/stablehlo/passes/quantization_patterns.h

          OperationState new_state(candidate_op->getLoc(),
                                   candidate_op->getName().getStringRef(), inputs,
                                   output_types, candidate_op->getAttrs());
          for (int i = 0; i < candidate_op->getNumRegions(); ++i) {
            new_state.addRegion();
          }
          Operation* quantized_op = rewriter.create(new_state);
          if (candidate_op->getNumRegions() != 0) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 10.9K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_with_tf2xla.cc

            encoding.getBounds()[i] == ShapedType::kDynamic) {
          return false;
        }
      }
      return true;
    }
    
    bool HasSymbolRefAttr(Operation* op) {
      for (const auto& attr : op->getAttrs()) {
        Attribute attr_value = attr.getValue();
        if (mlir::isa<SymbolRefAttr>(attr_value)) {
          return true;
        } else if (auto array_attr = mlir::dyn_cast<ArrayAttr>(attr_value)) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 21:49:50 UTC 2024
    - 9.7K bytes
    - Viewed (0)
Back to top