Search Options

Results per page
Sort
Preferred Languages
Advance

Results 71 - 80 of 86 for StringAttr (0.27 sec)

  1. tensorflow/compiler/mlir/tensorflow/utils/tpu_rewrite_device_util.cc

    }
    
    mlir::LogicalResult GetTopology(mlir::tf_device::ClusterOp cluster,
                                    std::string& topology) {
      mlir::StringAttr topology_attr =
          cluster->getAttrOfType<mlir::StringAttr>(tensorflow::kTopologyAttr);
      if (topology_attr) {
        topology = topology_attr.getValue();
        return mlir::success();
      } else {
        return cluster.emitOpError(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Jun 10 20:10:40 UTC 2024
    - 32.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/tensorflow/utils/tf_to_uniform_attribute_utils.cc

    }
    
    Attribute GetPaddingValue(PatternRewriter& rewriter,
                              llvm::StringMap<Attribute>& identifier_to_attr) {
      llvm::StringRef padding =
          mlir::dyn_cast<StringAttr>(identifier_to_attr["padding"]).getValue();
      return rewriter.getStringAttr(padding);
    }
    
    Attribute GetExplicitPaddingValue(
        PatternRewriter& rewriter, llvm::StringMap<Attribute>& identifier_to_attr) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 18.7K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/ir/tf_device.cc

    // tf_device.cluster_func
    //===----------------------------------------------------------------------===//
    
    LogicalResult ClusterFuncOp::verifySymbolUses(
        mlir::SymbolTableCollection& symbolTable) {
      StringAttr func_attr = getFuncAttr().getRootReference();
      func::FuncOp func =
          symbolTable.lookupNearestSymbolFrom<func::FuncOp>(*this, func_attr);
      if (!func) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 33.4K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc

        // Signature is not modified. We do not need to keep two copies.
        info.signature_change = false;
        if (lowered_callee != callee) {
          lowered_callee.setName(
              StringAttr::get(callee->getContext(), callee.getName()));
          callee.erase();
          SymbolTable(module).insert(lowered_callee);
        }
      } else {
        info.signature_change = true;
        for (auto& entry : callee_map) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 39.2K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/promote_resources_to_args.cc

        for (auto var_name_and_index : llvm::enumerate(var_handle_shared_names))
          function.setArgAttr(var_name_and_index.index() + var_handle_args_offset,
                              kResourceNameArgAttr,
                              StringAttr::get(context, var_name_and_index.value()));
      }
    }
    
    }  // namespace
    
    std::unique_ptr<OperationPass<ModuleOp>> CreatePromoteResourcesToArgsPass(
        llvm::ArrayRef<std::string> functions) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 17.1K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tfr/ir/tfr_ops.td

          for (auto& attr : (*this)->getAttrs()) {
            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;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 10:54:29 UTC 2024
    - 17.4K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/transforms/tensor_array_ops_decomposition.cc

          // Signature is not modified. We do not need to keep two copies.
          lowered_callee.setName(
              StringAttr::get(callee->getContext(), callee.getName()));
          callee.erase();
        } else {
          // Add the clone with a new name.
          lowered_callee.setName(StringAttr::get(
              callee->getContext(),
              llvm::formatv("{0}_tensorarray_decomposed", callee.getName()).str()));
        }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 02 20:41:19 UTC 2023
    - 40.2K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.cc

        // Terminators, qcast and decast are not quantizable.
        return false;
      }
    
      const bool attr_enforced_quantizable =
          op->hasAttrOfType<StringAttr>(kQuantTraitAttrName) &&
          op->getAttrOfType<StringAttr>(kQuantTraitAttrName).getValue().str() ==
              QuantTraitValues[QuantizationTrait::FullyQuantizable];
    
      const bool trait_enforced_quantizable =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 08 02:10:16 UTC 2024
    - 43.2K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/transforms/dilated_conv.h

      if (!op.getResult().hasOneUse()) {
        return rewriter.notifyMatchFailure(
            op, "result for current op has more than 1 use");
      }
      // Make sure Conv2D has 'VALID' padding.
      if (op->template getAttrOfType<StringAttr>("padding").getValue() != "VALID") {
        return rewriter.notifyMatchFailure(op,
                                           "Conv2D op doesn't have valid padding");
      }
      // Make sure dilations are all ones if set.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 20K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting_cleanup.cc

      if (func_uses.has_value() && llvm::hasSingleElement(func_uses.value()))
        return func;
      func::FuncOp cloned = func.clone();
      cloned.setPrivate();
      cloned.setName(
          StringAttr::get(func.getContext(), func.getName().str() + "_lifted"));
      SymbolTable(module).insert(cloned);
      return cloned;
    }
    
    // Eliminates unused results for If/Case operations. Also patches up the
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 19.7K bytes
    - Viewed (0)
Back to top