Search Options

Results per page
Sort
Preferred Languages
Advance

Results 121 - 130 of 181 for SmallVector (0.18 sec)

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

      return !device_attr || device_attr.getValue().empty();
    }
    
    llvm::StringMap<SmallVector<Cluster>> GetClusters(ModuleOp module) {
      TF::SideEffectAnalysis side_effect_analysis(module);
      auto main_func = module.lookupSymbol<func::FuncOp>("main");
      const TF::SideEffectAnalysis::Info& info =
          side_effect_analysis.GetAnalysisForFunc(main_func);
      llvm::StringMap<SmallVector<Cluster>> clusters = BuildAllClusters(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Apr 26 09:37:10 UTC 2024
    - 7.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/order_by_dialect.cc

        std::vector<std::pair<Operation*, Operation*>> side_effect_data;
        const detail::SideEffectAnalysisInfo* info = nullptr;
        auto extra_dependencies =
            [&](Operation* op,
                bool incoming) -> llvm::SmallVector<Operation*, 4> const& {
          return incoming ? info->DirectControlPredecessors(op)
                          : info->DirectControlSuccessors(op);
        };
        // Some tests have recursive calls and other shenanigans, so allow
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Nov 08 17:01:11 UTC 2022
    - 3.5K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/utils/attribute_utils.h

    #ifndef TENSORFLOW_COMPILER_MLIR_TENSORFLOW_UTILS_ATTRIBUTE_UTILS_H_
    #define TENSORFLOW_COMPILER_MLIR_TENSORFLOW_UTILS_ATTRIBUTE_UTILS_H_
    
    #include <string>
    #include <utility>
    
    #include "llvm/ADT/SmallVector.h"
    #include "mlir/IR/Attributes.h"  // from @llvm-project
    #include "mlir/IR/Operation.h"  // from @llvm-project
    #include "mlir/Support/LLVM.h"  // from @llvm-project
    #include "tensorflow/compiler/tf2xla/tf2xla_defs.h"
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 22 19:47:48 UTC 2024
    - 8.7K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/stablehlo/passes/convert_func_to_bfloat16.cc

        }
        if (isa<mlir::stablehlo::BitcastConvertOp>(op)) {
          // Skip `BitcastConvertOp`, which is handled by the other pattern.
          return failure();
        }
    
        // Update the results.
        SmallVector<Type, 4> new_results;
        if (failed(getTypeConverter()->convertTypes(op->getResultTypes(),
                                                    new_results)))
          return failure();
    
    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/lite/transforms/pin_ops_with_side_effects.cc

              return WalkResult::interrupt();
            }).wasInterrupted()) {
        fn.emitOpError("Can't have control ops in this pass.");
        signalPassFailure();
      }
    
      llvm::SmallVector<Operation *, 4> ops_with_side_effects;
    
      // We're iterating over all operations at the top block level, excluding
      // the return operation (which otherwise would be recognized as being
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 7K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/utils/perception_ops_utils.cc

      auto attrs = attr_.getAttrs();
      TfLitePoolParams pool_params;
    
      llvm::SmallVector<int32_t, 2> pool_size;
      if (failed(GetIntegerArraySafe(&func_, attrs, "pool_size", &pool_size, 2))) {
        return failure();
      }
      pool_params.filter_height = pool_size[0];
      pool_params.filter_width = pool_size[1];
    
      // Retrieve strides.
      llvm::SmallVector<int32_t, 2> strides;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 8.9K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/transforms/xla_call_module_serialization.cc

    // Converts `called_func` attributes in custom call ops back to `called_index`.
    FailureOr<ArrayAttr> DesymbolizeCustomCallCalledIndex(ModuleOp module) {
      Builder builder(module.getContext());
    
      SmallVector<Attribute> function_list;
      llvm::DenseMap<SymbolRefAttr, int> called_indexes;
    
      WalkResult result = module.walk([&](stablehlo::CustomCallOp op) {
        if (!IsTfFuncCustomCall(op)) {
          return WalkResult::advance();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 9.5K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/transforms/convert_tf_control_flow_to_scf.cc

      if (!destination_block->empty())
        rewriter.eraseOp(destination_block->getTerminator());
    
      destination_block->addArguments(
          block_arguments_type,
          SmallVector<Location>(block_arguments_type.size(),
                                source_block->getParent()->getLoc()));
      rewriter.mergeBlocks(source_block, destination_block,
                           destination_block->getArguments());
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Nov 03 12:35:38 UTC 2022
    - 9.4K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/utils/cluster_util.h

    // `get_target` that takes an op and returns the target name. `is_ignored_op` is
    // a hook to ignore certain ops that are not included in any clusters.
    llvm::StringMap<SmallVector<Cluster>> BuildAllClusters(
        Block& block, const TF::SideEffectAnalysis::Info& side_effect_analysis,
        std::function<std::string(Operation*)> get_target,
        std::function<bool(Operation*)> is_ignored_op);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Feb 21 22:33:23 UTC 2023
    - 2.7K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/quantization/tensorflow/ops/tf_quantize_op_test.cc

                          func::FuncDialect>();
      EmptyPatternRewriter pattern_rewriter(builder);
      Value value = CreateConstValue<float>(builder, module->getLoc(), {1024, 2},
                                            SmallVector<float>(2048, 0));
    
      QuantizationComponentSpec quant_spec;
      quant_spec.set_quantization_component(
          QuantizationComponentSpec::COMPONENT_WEIGHT);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sun Dec 10 05:52:02 UTC 2023
    - 3K bytes
    - Viewed (0)
Back to top