- Sort Score
- Result 10 results
- Languages All
Results 11 - 16 of 16 for getAxes (0.22 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/fuse_convolution_pass.cc
}); } // For dynamic case, the result of conv should be used by shape_of and mul. if (is_dynamic_broadcast) { auto conv_uses = (*conv_op.getODSResults(0).begin()).getUses(); if (std::distance(conv_uses.begin(), conv_uses.end()) != 2 || quant::FindUserOfType<shape::ShapeOfOp>(conv_op) == nullptr || quant::FindUserOfType<mhlo::MulOp>(conv_op) == nullptr) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 22:21:19 UTC 2024 - 8.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/cluster_formation.cc
for (const auto& p : llvm::zip(live_outs, launch_op.getResults())) { Value from = std::get<0>(p); // TODO(jingpu): move this to RegionUtils.h in MLIR core. for (auto& use : llvm::make_early_inc_range(from.getUses())) { if (launch_op_region->isAncestor(use.getOwner()->getParentRegion())) continue; use.set(std::get<1>(p)); } } } // Get all escaped live-out values of a region.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Oct 05 13:30:21 UTC 2023 - 6.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_hashtables.cc
mlir::isa<IntegerType>(key_dtype) && mlir::cast<IntegerType>(key_dtype).getWidth() == 64))) { return false; } for (auto& use : hashtable->getUses()) { Operation* user = use.getOwner(); // Allow consuming hash table ops that can be covered by TensorFlow Lite // hash table kernels.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/remove_unused_arguments.cc
// determine whether we have taken care of all uses and can remove // the arg. for (BlockArgument arg : func.getArguments()) { auto uses = arg.getUses(); use_count[arg.getArgNumber()] = std::distance(uses.begin(), uses.end()); argument_to_index.insert({arg, arg.getArgNumber()}); } // We're only considering return values that are the same (at this
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/odml_converter/transforms/outline_composites.cc
// if there is only one. Operation* GetUserIfOnlyOne(Operation* op) { if (op->getNumResults() != 1) return nullptr; auto result = op->getResult(0); if (!result.hasOneUse()) return nullptr; return (*result.getUses().begin()).getOwner(); } // Gets operation providing value for the given operand of given operation // if the given operation is the only user. Operation* GetInputOpWithOneUse(Operation* op, int opr_num) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 9.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/default_quant_params.cc
// bias is used immediately by the user. This assumption is always correct // after constant folding. bool UsedAsBias(Value value) { for (auto &use : value.getUses()) { auto biases = TFL::GetOpQuantSpec(use.getOwner())->biases_params; if (biases.find(use.getOperandNumber()) != biases.end()) return true; } return false; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.4K bytes - Viewed (0)