- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 54 for argTypes (0.27 sec)
-
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/quantization/tensorflow/passes/insert_restore_op.cc
kTfSavedModelIndexPathAttr, builder.getArrayAttr({file_prefix_attr}))}); const int insert_idx = func_op.getNumArguments(); func_op.insertArgument(insert_idx, /*argType=*/filename_op_type, arg_attrs, NameLoc::get(file_prefix_attr)); return func_op.getArgument(insert_idx); } // Creates a 1D string array constant for "tensor_names" input of `RestoreV2`
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Mar 12 06:02:20 UTC 2023 - 9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_array_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 Mar 02 20:41:19 UTC 2023 - 40.2K 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) -
platforms/core-configuration/model-core/src/main/java/org/gradle/model/internal/manage/schema/extract/ManagedProxyClassGenerator.java
methodVisitor.visitCode(); return methodVisitor; } private void putFirstMethodArgumentOnStack(MethodVisitor methodVisitor, Type argType) { int loadCode = argType.getOpcode(ILOAD); methodVisitor.visitVarInsn(loadCode, 1); } private void putFirstMethodArgumentOnStack(MethodVisitor methodVisitor) {
Registered: Wed Jun 12 18:38:38 UTC 2024 - Last Modified: Mon Dec 11 13:37:56 UTC 2023 - 55.3K 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)