- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for FunctionOpInterface (0.26 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/remove_unused_arguments.cc
if (!call) return; Region* region = call.getCallableRegion(); if (!region) return; // happens e.g. for external functions auto func = llvm::dyn_cast<FunctionOpInterface>(op.getOperation()); if (!func || do_not_touch.count(func)) return; llvm::BitVector unused_args(func.getNumArguments()); llvm::BitVector unused_results(func.getNumResults());
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/tfr/ir/tfr_ops.td
// Function related classes //===----------------------------------------------------------------------===// def TFR_TFRFuncOp : TFR_Op<"func", [HasParent<"ModuleOp">, FunctionOpInterface, IsolatedFromAbove, Symbol]> { let summary = "TFR Function defines a composition of other ops"; let description = [{
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 10:54:29 UTC 2024 - 17.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_main_function.cc
StringRef attr_name = attr.getName().getValue(); if (attr_name.starts_with("tf_saved_model.")) { func->removeAttr(attr_name); } } auto iface = cast<FunctionOpInterface>(func.getOperation()); for (int i = 0; i < func.getNumArguments(); ++i) { for (auto& attr : iface.getArgAttrs(i)) { const StringAttr& attr_name = attr.getName();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 16.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
int M = f.getNumArguments() - N; DCOMMENT_OP(op, "Inferring shape for with N = " << N << " and M = " << M); // Initialize with function input types. auto input_types = llvm::to_vector<1>( cast<FunctionOpInterface>(f.getOperation()).getArgumentTypes()); DatasetInput input_elements = GetDatasetInput(op->getOperand(0)); if (!input_elements) { op->emitWarning("unexpected dataset input; skipping function refinement");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0)