- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 17 for c_input_names (0.28 sec)
-
tensorflow/c/c_api.cc
for (int i = 0; i < ninputs; ++i) { input_names[i] = c_input_names[i]; } for (int i = 0; i < noutputs; ++i) { output_names[i] = c_output_names[i]; } for (int i = 0; i < ntargets; ++i) { target_oper_names[i] = c_target_oper_names[i]; } string new_handle; status->status = s->session->PRunSetup(input_names, output_names,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_main_function.cc
const std::string inputs_attr_str = mlir::cast<StringAttr>(inputs_attr).getValue().str(); std::vector<std::string> fn_input_names = absl::StrSplit(inputs_attr_str, ',', absl::SkipEmpty()); for (StringRef input_name : fn_input_names) { if (input_name_set.contains(input_name)) { // Found a duplicated name, all input names will be prefixed by
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 16.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/saved_model/saved_model.cc
} } for (auto func_name : func_names) { TFRTSavedModelSignatureInfo sig_info; sig_info.func_name = func_name; sig_info.input_names = input_names; sig_info.input_specs = input_specs; sig_info.input_devices = input_devices; sig_info.output_names = output_names; sig_info.output_specs = output_specs;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/python/jax_to_tfl_flatbuffer.cc
// Retrieve input names from model flags. std::vector<std::string> input_names; for (const auto& input : model_flags.input_arrays()) { input_names.push_back(input.name()); } const auto& inputs = absl::StrJoin(input_names, ","); mlir::OpBuilder builder(*module); llvm::SmallVector<mlir::NamedAttribute> attrs; attrs.push_back(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 11 19:29:56 UTC 2024 - 8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/cc/convert_asset_args.cc
} // namespace FailureOr<SmallVector<AssetFileDef>> ConvertAssetArgs(ModuleOp module_op) { func::FuncOp main_func_op = FindMainFuncOp(module_op); if (!main_func_op) return failure(); SmallVector<StringRef> input_names = GetEntryFunctionInputs(main_func_op); SymbolTable symbol_table(module_op); SmallVector<AssetFileDef> asset_file_defs; for (BlockArgument argument : main_func_op.getArguments()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/saved_model/saved_model.h
// be used to retrieve these information in a tf_saved_model module. struct TFRTSavedModelSignatureInfo { llvm::StringRef func_name; // The following are metadata for inputs. llvm::ArrayRef<llvm::StringRef> input_names; llvm::ArrayRef< std::pair<tensorflow::DataType, tensorflow::PartialTensorShape>> input_specs; llvm::ArrayRef<llvm::StringRef> input_devices; // The following are metadata for outputs.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jan 27 17:42:41 UTC 2024 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc
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()]); assert(!name.empty() && "expected non-empty name");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc
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()]); assert(!name.empty() && "expected non-empty name");
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/lite/python/saved_model_to_tfl_flatbuffer.cc
auto input_attr = tf_attrs.get("inputs"); if (!input_attr) { return errors::InvalidArgument("no inputs attribute found"); } auto input_names = mlir::cast<mlir::StringAttr>(input_attr).getValue(); input_names.split(function_input_names, ",", /*MaxSplit=*/-1, /*KeepEmpty=*/false); const int function_input_names_size = function_input_names.size();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun May 12 12:39:37 UTC 2024 - 11K bytes - Viewed (0) -
tensorflow/compiler/mlir/python/mlir.h
std::string ImportGraphDef(const std::string &proto, const std::string &pass_pipeline, bool show_debug_info, absl::string_view(input_names), absl::string_view(input_data_types), absl::string_view(input_data_shapes), absl::string_view(output_names), TF_Status *status);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 14 23:44:01 UTC 2023 - 5.6K bytes - Viewed (0)