- Sort Score
- Result 10 results
- Languages All
Results 121 - 130 of 181 for SmallVector (0.18 sec)
-
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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)