- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 31 for argTypes (0.18 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/passes/quantization_patterns.cc
func::FuncOp entry_func_op, TF::XlaCallModuleOp xla_call_module_op) { SmallVector<Type> arg_types; SmallVector<Location> arg_locs; for (const Value arg : xla_call_module_op.getArgs()) { arg_types.push_back(arg.getType()); arg_locs.push_back(arg.getLoc()); } SmallVector<Type> output_types;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 06:04:36 UTC 2024 - 41.7K bytes - Viewed (0) -
src/text/template/exec.go
} return v } // Special case for builtin and/or, which short-circuit. if isBuiltin && (name == "and" || name == "or") { argType := typ.In(0) var v reflect.Value for _, arg := range args { v = s.evalArg(dot, argType, arg).Interface().(reflect.Value) if truth(v) == (name == "or") { // This value was already unwrapped // by the .Interface().(reflect.Value). return v
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Fri May 24 21:22:24 UTC 2024 - 32K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc
}; // Updates func's type according to its current arguments and return values. void UpdateFuncType(func::FuncOp func) { llvm::SmallVector<Type, 8> arg_types; for (auto arg : func.getArguments()) arg_types.push_back(arg.getType()); func.setType( FunctionType::get(func.getContext(), arg_types, func.front().getTerminator()->getOperandTypes())); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.2K bytes - Viewed (0) -
tensorflow/compiler/jit/kernels/xla_ops.cc
OP_REQUIRES_OK_RETURN(ctx, std::vector<int>(), ctx->GetAttr("Tconstants", &constant_types)); DataTypeVector arg_types; OP_REQUIRES_OK_RETURN(ctx, std::vector<int>(), ctx->GetAttr("Targs", &arg_types)); int num_resources; OP_REQUIRES_OK_RETURN(ctx, std::vector<int>(), ctx->GetAttr("Nresources", &num_resources));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 41.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/c/c_api_unified_experimental_mlir.cc
ret_operands.push_back(operand->getValue()); } builder_.create<func::ReturnOp>(func_.getLoc(), ret_operands); auto arg_types = body.getArgumentTypes(); auto result_types = body.getTerminator()->getOperandTypes(); func_.setType(FunctionType::get(func_.getContext(), arg_types, result_types)); *f = new MlirFunction(std::move(context_), std::move(module_), func_); return absl::OkStatus(); } extern "C" {
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/tensorflow/transforms/tpu_space_to_depth_pass.cc
}; // Updates func argument type to have the updated input shape. void UpdateFuncType(func::FuncOp func) { auto arg_types = func.front().getArgumentTypes(); auto result_types = func.front().getTerminator()->getOperandTypes(); func.setType(FunctionType::get(func.getContext(), arg_types, result_types)); } void HandleFuncOp(Operation* op) { auto func = llvm::cast<func::FuncOp>(op); UpdateFuncType(func); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 29.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/unfreeze_constants.cc
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=*/{}); init_func->setAttr(kTfSavedModelExportedNamesAttr, builder.getStrArrayAttr( {"tf_saved_model.session_initializer_restore_op"}));
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/jit/encapsulate_xla_computations_pass_test.cc
EXPECT_EQ((DataTypeVector{DT_INT32, DT_FLOAT, DT_INT32, DT_FLOAT, DT_RESOURCE, DT_RESOURCE, DT_RESOURCE}), result.arg_types); EXPECT_EQ((DataTypeVector{DT_FLOAT, DT_INT32, DT_FLOAT, DT_FLOAT}), result.ret_types); TF_EXPECT_GRAPH_EQ(expected_body_def, result.gdef); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 16 18:03:15 UTC 2023 - 14.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/ir/tfr_ops.cc
} ParseResult TFRFuncOp::parse(OpAsmParser &parser, OperationState &result) { auto build_func_type = [](Builder &builder, ArrayRef<Type> arg_types, ArrayRef<Type> results, function_interface_impl::VariadicFlag, std::string &) { return builder.getFunctionType(arg_types, results); }; return function_interface_impl::parseFunctionOp( parser, result, /*allowVariadic=*/false,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Nov 21 16:55:41 UTC 2023 - 38.2K bytes - Viewed (0) -
src/runtime/heapdump.go
// locals. dumpbv(&bv, s.varp-uintptr(bv.n)*goarch.PtrSize-s.sp) } dumpint(fieldKindEol) // Record arg info for parent. child.argoff = s.argp - s.fp child.arglen = s.argBytes() child.sp = (*uint8)(unsafe.Pointer(s.sp)) child.depth++ stkmap = (*stackmap)(funcdata(f, abi.FUNCDATA_ArgsPointerMaps)) if stkmap != nil { child.args = stackmapdata(stkmap, pcdata) } else {
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Tue Apr 09 04:07:57 UTC 2024 - 17.6K bytes - Viewed (0)