- Sort Score
- Result 10 results
- Languages All
Results 71 - 80 of 86 for funcType (0.82 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_save_op.cc
/*shape=*/{}, /*elementType=*/builder.getType<TF::StringType>()); FunctionType func_type = builder.getFunctionType( /*inputs=*/{filename_input_type}, /*results=*/{}); auto save_func = builder.create<func::FuncOp>( NameLoc::get(builder.getStringAttr(kTfQuantSaveFuncName)), /*sym_name=*/kTfQuantSaveFuncName, func_type); save_func.addEntryBlock(); save_func.setPrivate(); return save_func; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 05:52:39 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/lstm_utils_test.cc
projection_type, layer_norm_scale_type}; auto func_type = builder->getFunctionType(input_types, output_type); auto func = func::FuncOp::create( mlir::NameLoc::get(builder->getStringAttr("fused_func")), "fused_func", func_type, {}); func.addEntryBlock(); std::vector<std::string> attributes; if (ln) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/promote_resources_to_args.cc
func::FuncOp function, bool add_validation, llvm::SmallVectorImpl<std::string>* var_handle_shared_names) { Block& block = function.front(); auto func_type = function.getFunctionType(); auto func_arg_types = llvm::to_vector<4>(func_type.getInputs()); llvm::SmallDenseMap<llvm::StringRef, int> var_arg_index_by_name; for (auto var_handle_op : llvm::make_early_inc_range(block.getOps<TF::VarHandleOp>())) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/ops/tf_quantize_op.cc
CreateUnknownShapeFromElementType(result_type); FunctionType func_type = FunctionType::get(rewriter.getContext(), {create_unknown_input_shape}, {create_unknown_output_shape}); func::FuncOp quantization_func = rewriter.create<func::FuncOp>(input_op->getLoc(), func_name, func_type); OpBuilder::InsertionGuard guard = OpBuilder::InsertionGuard(rewriter);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc
// them in the function signature. FunctionType func_type = rewriter.getFunctionType({}, {}); func::FuncOp func = rewriter.create<func::FuncOp>( loc, kImportModelDefaultGraphFuncName, func_type); rewriter.setInsertionPointToStart(func.addEntryBlock()); auto executor_graph = rewriter.create<tf_executor::GraphOp>(loc, func_type.getResults());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/cluster_tf_ops_pass.cc
std::replace(func_name.begin(), func_name.end(), '/', '_'); FunctionType func_type = FunctionType::get(context, input_types, result_types); Location loc = metadata.ops.front()->getLoc(); func::FuncOp func_op = func::FuncOp::create(loc, func_name, func_type); // Sets the device attribute for every input and every result of the // function.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 13.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/unfreeze_constants.cc
builder.setInsertionPointAfter(session_init_op); const Location loc = builder.getUnknownLoc(); const auto func_type = builder.getFunctionType(/*inputs=*/{}, /*results=*/{}); auto init_func = builder.create<func::FuncOp>( loc, /*sym_name=*/"init_func_restore_op", func_type); builder.createBlock(&init_func.getBody(), /*insertPt=*/init_func.begin(), /*arg_types=*/{}, /*arg_locs=*/{});
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 16 15:04:53 UTC 2023 - 14K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_main_function.cc
arg_locs.push_back(arg.getLoc()); } } // Creates a new main function. auto func_type = FunctionType::get(context, arg_types, result_types); auto main_func = builder.create<func::FuncOp>( module_op.getLoc(), kImportModelDefaultGraphFuncName, func_type); builder.createBlock(&main_func.getBody(), main_func.begin(), arg_types, arg_locs);
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/sparsecore/embedding_pipelining.cc
Append(result_types, callers.core_tpu->getResultTypes()); // Create the function based on input and result types and values. auto func_type = mlir::FunctionType::get(builder.getContext(), input_types, result_types); func::FuncOp then_func = func::FuncOp::create(loc, name, func_type); then_func.setPrivate(); symbol_table.insert(then_func); mlir::OpBuilder func_builder =
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/quantization/common/lift_as_function_call.cc
TypeRange result_types{ValueRange{results}}; auto func_type = FunctionType::get(context, arg_types, result_types); SmallVector<Location> arg_locs; for (Value arg : arguments) { arg_locs.push_back(arg.getLoc()); } auto wrap_func = builder.create<func::FuncOp>(location, func_name, func_type); wrap_func.setVisibility(SymbolTable::Visibility::Private);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.8K bytes - Viewed (0)