- Sort Score
- Result 10 results
- Languages All
Results 51 - 60 of 85 for function_type (0.27 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/odml_converter/transforms/outline_composites.cc
PatternRewriter& rewriter, Block* insertion_point) { rewriter.setInsertionPointToStart(insertion_point); auto ftype = FunctionType::get(rewriter.getContext(), {type}, {type}); auto name = rewriter.getStringAttr("gelu_decomp"); func::FuncOp new_func = rewriter.create<func::FuncOp>( insertion_point->front().getLoc(), name, ftype);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 9.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_save_op.cc
builder.setInsertionPointToEnd(&module_op.getBodyRegion().front()); auto filename_input_type = RankedTensorType::get( /*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)),
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/quantization/tensorflow/passes/preprocess_op.cc
new_float_func_args[weight_operand_idx] = reshape_op; new_float_func.getArgument(weight_operand_idx).setType(new_shape); new_float_func.setType(FunctionType::get( getContext(), TypeRange{ValueRange{new_float_func_args}}, new_float_func.getResultTypes())); symbol_table.insert(new_float_func);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.4K bytes - Viewed (0) -
src/go/internal/gccgoimporter/parser.go
return types.NewTuple(types.NewParam(token.NoPos, pkg, "", taa)) case '(': params, _ := p.parseParamList(pkg) return params default: return nil } } // FunctionType = ParamList ResultList . func (p *parser) parseFunctionType(pkg *types.Package, nlist []any) *types.Signature { t := new(types.Signature) p.update(t, nlist) params, isVariadic := p.parseParamList(pkg)
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Fri Feb 02 23:14:07 UTC 2024 - 31.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc
xla_input_shapes->clear(); mlir::func::FuncOp main_func = module.lookupSymbol<mlir::func::FuncOp>("main"); TF_RET_CHECK(main_func != nullptr) << "No main function found"; mlir::FunctionType func_type = main_func.getFunctionType(); int num_args = func_type.getNumInputs(); xla_input_shapes->reserve(num_args); std::vector<xla::Shape> individual_arg_shapes;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 17:24:39 UTC 2024 - 45.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_pipelining.cc
// Note, for a while loop body function, the operand types and result types // are identical. auto body_func_type = mlir::FunctionType::get( &getContext(), new_while_operand_types, new_while_operand_types); auto cond_func_type = mlir::FunctionType::get( &getContext(), new_while_operand_types, orig_cond_func.getResultTypes()); func::FuncOp cond =
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/tensorflow/transforms/shape_inference.cc
.replaceAllUsesWith( builder.clone(*operand.getDefiningOp())->getResult(0)); } } // Update/use input shapes for function. FunctionType func_type = func.getFunctionType(); func.setType(FunctionType::get(func.getContext(), op.getOperandTypes(), func_type.getResults())); // Run shape inference on the function. if (failed(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/lstm_utils.cc
auto output_type = tensorflow::GetTypeFromTFTensorShape( output_shape, mlir::cast<RankedTensorType>(input_.getType()).getElementType()); fused_func_op_.setType(mlir::FunctionType::get(fused_func_op_.getContext(), input_types, output_type)); } LogicalResult ConvertLSTMCellSimpleToFusedLSTM::RewriteFunc() { LogicalResult result = Initialize();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 36.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
// unranked tensors and could be ranked tensors in cases where result type // depends on an op with static output shape like tf.Const. absl::StatusOr<mlir::FunctionType> InferLibFunctionType( const FunctionBody& fbody); // Extracts arg and ret nodes from FunctionBody. void GetArgsAndRetsFromFunctionBody( const FunctionBody& fbody,
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/functional_control_flow_to_cfg.cc
// that is compatible for tensor cast. static Operation* CallFn(Location loc, const std::function<Value(int)>& get_arg, func::FuncOp fn, OpBuilder* builder) { FunctionType fn_type = fn.getFunctionType(); llvm::SmallVector<Value, 4> operands; int num_operands = fn_type.getNumInputs(); operands.reserve(num_operands); for (int i = 0; i < num_operands; ++i) { Value val = get_arg(i);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jan 13 11:42:59 UTC 2023 - 12.2K bytes - Viewed (0)