- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for GetNumInputs (1.24 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
} auto func_type = func.getFunctionType(); if (func_type.getNumInputs() != 2) { return op.emitOpError() << "expects reduction function to take 2 parameters, but " "has " << func_type.getNumInputs() << " parameter(s)"; } return success(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_executor.cc
// input and the outputs are all using this type and predicate is tensor<i1> // type). if (mlir::isa<FunctionType>(types.front())) { FunctionType type = mlir::cast<FunctionType>(types.front()); if (type.getNumInputs() < 2) return parser.emitError(parser.getNameLoc()) << " expects a single data type and a predicate"; result.types.assign(type.getResults().begin(), type.getResults().end());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 42.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/tftext_utils.cc
builder.create<func::ReturnOp>(func.getLoc(), op.getResults()); return success(); } LogicalResult VerifySgnnProjection(func::FuncOp func, FuncAttr attr) { if (func.getFunctionType().getNumInputs() != 2 || func.getFunctionType().getNumResults() != 1) { return func.emitError() << "Mismatched number of inputs and outputs."; } auto values_type = GetInputType(func, 0);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
for (func::FuncOp func : {op.cond_function(), op.body_function()}) { ++func_index; if (!func) continue; FunctionType func_type = func.getFunctionType(); int num_inputs = func_type.getNumInputs(); int num_results = func_type.getNumResults(); // For each argument type in function's arguments, change it to uranked // tensor type if it's a variant type.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 70.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc
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; individual_arg_shapes.reserve(num_args); for (int i = 0; i < num_args; ++i) {
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/shape_inference.cc
FunctionType func_type = func.getFunctionType(); bool needs_refinement = false; SmallVector<Type, 4> new_arg_types; new_arg_types.reserve(func_type.getNumInputs()); // Update argument types in-place using the provided arg_shapes. for (size_t i = 0; i < func_type.getNumInputs(); ++i) { ArrayRef<int64_t> shape = arg_shapes[i]; Type element_type; if (auto input_ty =
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/tensorflow/ir/tf_ops_a_m.cc
if (!func) { return emitError("'f' attribute refers to an undefined function: ") << func_name; } FunctionType func_ty = func.getFunctionType(); int func_arg_count = func_ty.getNumInputs(); int arg_count = getArgs().size(); if (arg_count != func_arg_count) { return emitError() << "argument count mismatch: 'args' has " << arg_count
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 146.7K bytes - Viewed (0)