- Sort Score
- Result 10 results
- Languages All
Results 51 - 57 of 57 for getFunctionType (0.25 sec)
-
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)