- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for getNumArguments (0.67 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/decompose_optionals.cc
if (!symbol) return failure(); auto f = llvm::dyn_cast<mlir::func::FuncOp>(symbol); if (call.getOperation()->getNumOperands() != f.getBody().front().getNumArguments()) { return failure(); // RemoteCall et al } rewriter.startOpModification(f); bool changed = false; for (auto [call_arg, body_arg] :
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 9.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_tf_graph.cc
constexpr char kAliasingAttr[] = "tf.aliasing_output"; llvm::SmallDenseMap<unsigned, unsigned> output_to_input_alias; unsigned num_arguments = main_fn.getNumArguments(); for (unsigned arg_index = 0; arg_index < num_arguments; ++arg_index) { if (auto aliasing_output = main_fn.getArgAttrOfType<mlir::IntegerAttr>( arg_index, kAliasingAttr))
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 14K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc
unique_output_names.push_back( exporter.op_to_name_.GetUniqueName(tensor_id_node)); } } if (!input_names.empty()) { TF_RET_CHECK(input_names.size() == block.getNumArguments()); for (const auto& it : llvm::enumerate(function.getArguments())) { // TODO(lyandy): Update when changing feed/fetch import. std::string name(input_names[it.index()]);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 23:04:51 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
} bool ShapeInference::InferShapeForDatasetOpCommon(Operation* op, FuncOp f, int64_t max_iterations) { int N = op->getNumOperands() - 1; int M = f.getNumArguments() - N; DCOMMENT_OP(op, "Inferring shape for with N = " << N << " and M = " << M); // Initialize with function input types. auto input_types = llvm::to_vector<1>(
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/flatbuffer_export.cc
mlir::dyn_cast_or_null<mlir::StringAttr>(dict_attr.get("inputs"))) { str.getValue().split(input_names, ',', /*MaxSplit=*/-1, /*KeepEmpty=*/false); if (input_names.size() != fn.getNumArguments()) { fn.emitWarning() << "invalid entry function specification"; return; } for (const auto& it : llvm::enumerate(fn.getArguments())) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
updated_result_types)); }); Region &entry = func.getRegion(); TypeConverter::SignatureConversion signature_conversion( entry.getNumArguments()); for (const BlockArgument &arg : entry.getArguments()) { signature_conversion.addInputs(arg.getArgNumber(), updated_argument_types[arg.getArgNumber()]); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 70.7K bytes - Viewed (0)