- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 89 for getOperands (0.16 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/annotate_parameter_replication.cc
mlir::cast<IntegerAttr>(mirrored_index).getInt()); } } auto func = llvm::cast<func::FuncOp>(m.lookupSymbol(cluster_func.getFunc())); for (auto entry : llvm::enumerate(cluster_func.getOperands())) { auto operand = SkipIdentityAndReadVariable(entry.value()); auto block_arg = mlir::dyn_cast<BlockArgument>(operand); if (block_arg && block_arg.getOwner() == &replicate.GetBody()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_outline_tpu_island.cc
func_operand_types.push_back(operand.getType()); // Function results are the yield operands SmallVector<Type, 16> func_result_types; for (Value operand : island_op.GetYield().getOperands()) func_result_types.push_back(operand.getType()); FunctionType func_type = FunctionType::get(ctx, func_operand_types, func_result_types); // Create the outlined function
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/cast_bf16_ops_to_f32.cc
LogicalResult match(Operation* op) const override { if (isa<TF::CastOp, TF::ConstOp>(op) || op->getName().hasTrait<OpTrait::ZeroOperands>()) { return failure(); } for (Value input : op->getOperands()) { if (getElementTypeOrSelf(input).isBF16()) { return success(); } } for (Value value : op->getResults()) { if (getElementTypeOrSelf(value).isBF16()) { return success();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Dec 10 05:52:02 UTC 2023 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/convert_launch_func_to_tf_call.cc
module.walk([&](tf_device::LaunchFuncOp launch) { OpBuilder builder(launch); auto call_op = builder.create<TF::PartitionedCallOp>( module.getLoc(), launch.getResultTypes(), launch.getOperands(), SymbolRefAttr::get(builder.getContext(), launch.getFunc()), /*config=*/builder.getStringAttr(""), /*config_proto=*/builder.getStringAttr(""), /*executor_type=*/builder.getStringAttr(""));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 31 21:08:09 UTC 2023 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/xla_rewrite.cc
OpBuilder &builder) { llvm::SmallVector<Value> non_resource_args, resource_args; bool has_resources = false, in_order = true; for (const Value &arg : cluster_func_op.getOperands()) { if (!mlir::isa<TF::ResourceType>(getElementTypeOrSelf(arg.getType()))) { non_resource_args.push_back(arg); if (has_resources) in_order = false; } else { resource_args.push_back(arg);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/analysis/tf_dataflow.h
} else if (auto while_op = dyn_cast<TF::WhileRegionOp>(op)) { for (auto ®ion : while_op->getRegions()) { for (auto [arg, value] : llvm::zip(region.getArguments(), while_op->getOperands())) { this->join(getLatticeElement(arg), *getLatticeElement(value)); } } } else if (auto while_op = dyn_cast<TF::WhileOp>(op)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 06 23:53:00 UTC 2024 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/fake_quant_utils.cc
// Recreate the operation by using the wrapper's operands and return types. // TODO(fengliuai): copy the regions. OperationState state(op->getLoc(), op->getName().getStringRef(), parent_op->getOperands(), parent_op->getResultTypes(), op->getAttrs(), op->getSuccessors()); Operation* inlined = builder.create(state); parent_op->replaceAllUsesWith(inlined); parent_op->erase(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 03 00:14:05 UTC 2023 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/smuggle_disallowed_ops.cc
SmallVector<NamedAttribute> attrs{op->getAttrs()}; attrs.push_back(call_target); auto custom_call = rewriter.create<mlir::stablehlo::CustomCallOp>( op->getLoc(), op->getResultTypes(), op->getOperands(), attrs); rewriter.replaceOp(op, custom_call.getResults()); return success(); } } // namespace template <typename OpTy> class SmuggleOpPattern : public OpRewritePattern<OpTy> { public:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 19 01:14:04 UTC 2024 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/xla_rewrite_util.cc
if (llvm::isa<mlir::TF::TPUPartitionedOutputV2Op>(user)) { assert(user->use_empty()); user->erase(); } } } for (auto operand : cluster.getOperands()) { mlir::Operation* def = operand.getDefiningOp(); if (operand.hasOneUse() && llvm::isa_and_nonnull<mlir::TF::TPUPartitionedInputV2Op>(def)) { operand.dropAllUses(); def->erase();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 13 03:57:18 UTC 2023 - 4.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_with_tf2xla.cc
// while creating the ops so make sure there aren't any type changes between // the original op operands and the operands during the conversion. for (auto&& [old_val, new_val] : llvm::zip(op->getOperands(), operands)) { if (old_val.getType() != new_val.getType()) return failure(); } auto abstractOp = op->getRegisteredInfo(); if (!abstractOp) return failure();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 9.7K bytes - Viewed (0)