- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 126 for func_20 (0.52 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_program_key.cc
// tf_device.replicate. Operation* FindCompilePredecessor(Operation* func_op, Operation* preprocess_op) { llvm::DenseSet<Block*> blocks = GetAllBlocksBetween(preprocess_op, func_op); llvm::DenseMap<Block*, Operation*> scope; Operation* o = preprocess_op; while (o && o != func_op) { scope[o->getBlock()] = o; o = o->getParentOp(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/lstm_utils.cc
Value input = func_op.getArgument(0); Value output_init_state = func_op.getArgument(1); Value hidden_init_state = func_op.getArgument(2); Value weight_kernel = func_op.getArgument(3); Value recurrent_kernel = func_op.getArgument(4); Value bias = func_op.getArgument(5); // The func op should have 5 outputs. if (func_op.getNumResults() != 5) return failure();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 36.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/unwrap_xla_call_module_op.cc
if (call_op_has_platform_index_arg) { arg_mapper.map(func_op.getArgument(0), builder.create<mhlo::ConstantOp>( func_op.getLoc(), builder.getI16IntegerAttr(0))); } for (auto [func_arg, operand] : llvm::zip_equal( func_op.getArguments().take_back(call_op.getNumOperands()), call_op.getOperands())) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 07:39:40 UTC 2024 - 4.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/analysis/test_tensor_array_side_effect_analysis.cc
TensorArraySideEffectAnalysis tensor_array_side_effect_analysis(module); for (auto func_op : module.getOps<mlir::func::FuncOp>()) { func_op.emitRemark() << "HasAtMostTensorArrayEffect: " << tensor_array_side_effect_analysis .HasAtMostTensorArrayEffect(func_op); } } }; mlir::PassRegistration<TestTensorArraySideEffectAnalysis> pass;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Apr 10 21:32:05 UTC 2022 - 1.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/remove_sharding_custom_call.cc
}; void RemoveShardingCustomCallPass::runOnOperation() { func::FuncOp func_op = getOperation(); MLIRContext& ctx = getContext(); RewritePatternSet patterns(&ctx); populateWithGenerated(patterns); FrozenRewritePatternSet frozen_patterns(std::move(patterns)); if (failed(applyPatternsAndFoldGreedily(func_op, frozen_patterns))) { func_op.emitWarning() << "Failed to converge "
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 13 07:04:47 UTC 2024 - 2.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/convert_to_legacy_compile_and_replicate_attributes.cc
ConvertToLegacyCompileAndReplicateAttributesPass> { void runOnOperation() override; }; LogicalResult ConvertToLegacyAttributes(func::FuncOp func_op) { auto result = func_op->walk([&](mlir::Operation* op) { if (failed(TF::HasValidCompilationAndReplicationAttributes(*op))) return WalkResult::interrupt(); if (op->hasAttr(TF::kReplicationInfoAttr)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/manipulate_model_attr.cc
void AddEntryFunctionInput(StringRef input_name, func::FuncOp func_op) { auto entry_func_attr = func_op->getAttrOfType<DictionaryAttr>(kTfEntryFunctionAttr); if (!entry_func_attr) return; auto entry_func_attrs = SmallVector<NamedAttribute>(entry_func_attr.begin(), entry_func_attr.end()); MLIRContext* ctx = func_op.getContext(); for (auto& named_attr : entry_func_attrs) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 26 01:13:26 UTC 2023 - 2.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_device_copy_conversion.cc
}; // Folds tf.IdentityOp and tf.IdentityNOp if op device and the argument devices // from the defining ops match. void TensorDeviceCopyConversionPass::runOnOperation() { func::FuncOp func_op = getOperation(); auto should_fold_op_func = [&func_op](const Value &arg, const StringAttr &op_device) { // In TFRT TPU, tensor transfer is handled specifically by D2H and
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/nms_utils.cc
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); Value score_threshold = func_.getArgument(4); auto output_type0 = func_.getFunctionType().getResult(0);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/analysis/cost_analysis.h
public: explicit CostAnalysis( mlir::func::FuncOp func_op, const tfrt_stub::CostRecorder* cost_recorder = nullptr) { cost_recorder_ = cost_recorder; AnalyzeArguments(func_op); AnalyzeBlock(&func_op.front()); } int64_t GetCost(mlir::Operation* op) const; private: void AnalyzeArguments(mlir::func::FuncOp func_op); void AnalyzeBlock(mlir::Block* block);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 06 03:08:33 UTC 2023 - 3.1K bytes - Viewed (0)