- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 73 for call_op_ (0.14 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/compose_uniform_quantized_type_pass.cc
return GetInverseScalesConstantOp().getValue(); } func::CallOp GetCallOp() { return call_op_; } FlatSymbolRefAttr GetFunction() { return call_op_.getCalleeAttr(); } private: explicit UniformQuantizeFunctionCallPattern(func::CallOp call_op) : call_op_(call_op) {} func::CallOp call_op_; }; // Matches the following pattern that represents uniform dequantization. //
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 64.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/xla_cluster_formation.cc
}; void EncapsulatePartitionedCall(Operation *call_op, mlir::StringAttr callee_name) { OpBuilder builder(call_op); auto cluster = builder.create<mlir::tf_device::ClusterOp>( call_op->getLoc(), call_op->getResultTypes()); cluster.getBody().push_back(new Block); call_op->replaceAllUsesWith(cluster.getResults()); call_op->moveBefore(&cluster.GetBody(), cluster.GetBody().end());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 19 19:09:44 UTC 2023 - 6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/common/outline_operations.cc
// for the call_op. Operation* last_output = subgraph.partition_ops_.back(); builder.setInsertionPoint(last_output); auto call_op = builder.create<func::CallOp>(last_output->getLoc(), func, subgraph.FuncArguments()); ops_added.call_op = call_op; // FuncOutputs refer to the original `Values` in input module which are now
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/merge_fusion_with_dequantize.cc
: public OpRewritePattern<func::CallOp> { public: explicit MergeFusionWithUniformDequantizePattern(MLIRContext* context) : OpRewritePattern<func::CallOp>(context) {} LogicalResult matchAndRewrite(func::CallOp call_op, PatternRewriter& rewriter) const override { if (call_op.getNumResults() != 1) return failure(); auto users = call_op->getUsers(); for (auto user : users) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/report.cc
std::optional<QuantizationResult> GetQuantizationResult(func::CallOp call_op) { const StringRef callee_name = call_op.getCalleeAttr().getValue(); if (!callee_name.starts_with(kQuantizedFuncPrefix)) { return std::nullopt; // `call_op` is not a quantized function call. } absl::StatusOr<Method> method = GetQuantizationMethod(call_op); if (!method.ok()) { call_op->emitError() << "Failed to get quantization method: "
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/raise_to_tf.cc
PatternRewriter& rewriter, CallOp call_op, const SmallVectorImpl<Type>& output_types, const SmallVectorImpl<Value>& inputs, const NamedAttrList& attr_list, const llvm::StringMap<Attribute>& derived_attrs) const { // Create the new op Location loc = call_op.getLoc(); rewriter.setInsertionPointAfter(call_op); std::string tf_op_name = GetTFOpName(call_op.getCallee());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/quantization_patterns.cc
// Checks if an op calls a composite function and all the inputs and outputs are // quantized. bool IsQuantizedCompositeFunction(func::CallOp call_op) { if (!call_op.getCallee().starts_with("quantized_")) { return false; } bool has_quantized_types = false; for (Value operand : call_op.getOperands()) { if (const TensorType type = mlir::dyn_cast<TensorType>(operand.getType())) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 06:04:36 UTC 2024 - 41.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/decompose.cc
bool changed = false; auto walk_result = func.walk([&](CallOp call_op) { auto callee = table.lookup<TFRFuncOp>(call_op.getCallee()); if (!callee || callee.isExternal()) return WalkResult::advance(); // Record the boundary of the inlined operations. The inlined operation will // be inserted between these two operations. Operation* inlined_point = call_op.getOperation(); Operation* after_inlined_point =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/add_functions_for_exported_names.cc
other.addEntryBlock(); OpBuilder builder(other.getRegion()); auto call_op = builder.create<mlir::func::CallOp>( f.getLoc(), f.getFunctionType().getResults(), f.getSymName(), other.getRegion().getArguments()); builder.create<mlir::func::ReturnOp>(f.getLoc(), call_op.getResults()); } Unexport(f); } } } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 19 08:06:04 UTC 2023 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/unwrap_xla_call_module_op.cc
}; void UnwrapXlaCallModuleOp(TF::XlaCallModuleOp call_op, SymbolTable& symbol_table) { // Do not inline lifted quantized functions used for fusing patterns. // TODO - b/310539922: Remove reference to TF/TFL utils. if (call_op->hasAttr(kQuantTraitAttrName)) { return; } auto function_name = call_op ->getAttrOfType<FlatSymbolRefAttr>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 07:39:40 UTC 2024 - 4.8K bytes - Viewed (0)