- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for getAllResultAttrs (0.26 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tpu_device_propagation.cc
// we have many result values. llvm::SmallVector<std::vector<NamedAttribute>, 8> result_attrs; { llvm::SmallVector<DictionaryAttr, 8> tmp; func.getAllResultAttrs(tmp); for (const auto& res : tmp) { result_attrs.push_back(res.getValue().vec()); } } mlir::Builder builder(func.getOperation());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc
getContext(), ftype.getInputs(), graph_func.getAllArgAttrs(), arg_types, arg_attrs)) || failed(FilterTfgSpecificArgResultAttributes( getContext(), ftype.getResults(), graph_func.getAllResultAttrs(), res_types, res_attrs))) return failure(); // Update the function type which has excluded the control args. func->setAttr("function_type", TypeAttr::get(rewriter.getFunctionType(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
// Fetch inputs and outputs from the signature. llvm::SmallVector<mlir::DictionaryAttr, 4> arg_attrs, res_attrs; main_op.getAllArgAttrs(arg_attrs); main_op.getAllResultAttrs(res_attrs); std::vector<std::string> sig_def_inputs = GetStringsFromDictionaryAttr(arg_attrs, kTfSavedModelIndexPathAttr); std::vector<std::string> sig_def_outputs =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0)