- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 50 for getFunctionType (0.43 sec)
-
tensorflow/compiler/mlir/tensorflow/c/c_api_unified_experimental_mlir.cc
// TODO(aminim) figure out the location story here module_ = ModuleOp::create(builder_.getUnknownLoc()); func_ = func::FuncOp::create( builder_.getUnknownLoc(), name, builder_.getFunctionType(std::nullopt, std::nullopt)); module_->push_back(func_); builder_ = OpBuilder::atBlockBegin(func_.addEntryBlock()); } void Release() override { delete this; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/quantization_patterns.cc
} SmallVector<Type> output_types; for (const Value output : xla_call_module_op.getOutput()) { output_types.push_back(output.getType()); } entry_func_op.setFunctionType( rewriter.getFunctionType(arg_types, output_types)); // Replace argument types and locs. Block& entry = entry_func_op->getRegion(0).front(); for (auto [arg, arg_type, arg_loc] :
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/tensorflow/translate/import_model.cc
auto* node = graph_->FindNodeId(ret->id()); TF_ASSIGN_OR_RETURN(auto type, InferInputType(*node, /*idx=*/0, builder)); ret_types.push_back(type); } return builder.getFunctionType(arg_types, ret_types); } // Stateful helper class to import a TensorFlow model expressed in GraphDef into // an MLIR Module. // // The nodes defined in the graph are converted to a function called
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_sequencing.cc
builder.setInsertionPointToEnd(&module.getBodyRegion().back()); func::FuncOp func_op = builder.create<func::FuncOp>( module.getLoc(), name, builder.getFunctionType(input_types, output_types)); func_op.setPrivate(); return func_op; } TF::StatefulPartitionedCallOp EncapsulateOpsInFunc( OpBuilder& builder, const llvm::SetVector<Operation*>& ops,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_composite_functions.cc
// DCE pass. OpBuilder builder(module_); builder.setInsertionPointToEnd(&module_.getBodyRegion().back()); const auto func_type = builder.getFunctionType(/*inputs=*/{}, /*results=*/{}); auto summary_func = builder.create<func::FuncOp>( builder.getUnknownLoc(), /*sym_name=*/"summary", func_type); summary_func.setPrivate();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 54.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
auto module = op->getParentOfType<ModuleOp>(); auto function = dyn_cast_or_null<func::FuncOp>(SymbolTable::lookupSymbolIn(module, func)); FunctionType function_ty = function.getFunctionType(); for (auto arg_in : llvm::zip(args, function_ty.getInputs())) { if (std::get<0>(arg_in).getType() != std::get<1>(arg_in)) { // Argument type and input type mismatch. return false; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc
llvm::SmallVector<Type, 4> output_types; output_types.reserve(outputs.size()); for (Value v : outputs) output_types.emplace_back(v.getType()); auto func_type = builder->getFunctionType(operand_types, output_types); FuncOp outlined_func = FuncOp::create(ops.front()->getLoc(), kHostFunctionAttr, func_type); // Create function body.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 68.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
return op->emitOpError() << "expects " << branch_name(branch.index()) << " (" << branch.value() << ") to point to a defined function"; FunctionType branch_type = branch_func.getFunctionType(); std::string desc = branch_name(branch.index()) + " input"; TypeRangeWithDesc branch_input{branch_type.getInputs(), desc}; if (failed(VerifyTypeRangesAreCompatible(op, branch_input, input)))
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 146.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_pipelining.cc
auto out_types = GetValueTypes(outputs); builder.setInsertionPointToEnd(&module.getBodyRegion().back()); auto func_op = builder.create<func::FuncOp>( module.getLoc(), name, builder.getFunctionType(in_types, out_types)); func_op.setPrivate(); symbol_table.insert(func_op); return func_op; } TF::StatefulPartitionedCallOp EncapsulateOpsInFunc( OpBuilder& builder, SymbolTable& symbol_table,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 92.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
} mlir::func::FuncOp callee = module_op_->lookupSymbol<mlir::func::FuncOp>( mlir::cast<FlatSymbolRefAttr>(op.getCalledComputations()[0])); mlir::FunctionType callee_type = callee.getFunctionType(); SmallVector<Type, 4> expected_callee_input_types; auto num_inputs = op.getInputs().size() / 2; for (unsigned i = 0; i < num_inputs; ++i) { auto input_type =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0)