- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for output_arrays (0.18 sec)
-
tensorflow/compiler/mlir/lite/python/saved_model_to_tfl_flatbuffer.cc
for (const auto& output_array : model_flags.output_arrays()) { if (function_output_names_set.count(output_array) == 0) { return errors::InvalidArgument("output array name (", output_array, ") does not exist in the given graph"); } } return absl::OkStatus(); } Status ConvertSavedModelToTFLiteFlatBuffer(
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/tensorflow/translate/tf_mlir_translate.cc
const std::vector<std::string>& output_arrays, const std::vector<std::string>& control_output_arrays, const GraphdefToMlirOptions& import_options, mlir::MLIRContext* context) { auto module_or = GraphdefToMlirImport( input, input_arrays, input_dtypes, input_shapes, output_arrays, control_output_arrays, import_options, context); if (!module_or.status().ok()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 11:51:44 UTC 2024 - 14.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/tf_mlir_translate.h
llvm::StringRef input, const std::vector<std::string>& input_arrays, const std::vector<std::string>& input_dtypes, const std::vector<std::optional<std::vector<int>>>& input_shapes, const std::vector<std::string>& output_arrays, const std::vector<std::string>& control_output_arrays, const GraphdefToMlirOptions& import_options, mlir::MLIRContext* context); ABSL_DEPRECATED(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 5.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tf_to_tfl_flatbuffer.h
// source into a MLIR module. If `input_mlir` is true, load from a MLIR source // file; otherwise, load from a GraphDef. // Setting prune_unused_nodes to true, would prune unreachable nodes if // output_arrays is specified. absl::StatusOr<mlir::OwningOpRef<mlir::ModuleOp>> LoadFromGraphdefOrMlirSource( const std::string& input_filename, bool input_mlir, bool use_splatted_constant, const std::vector<std::string>& extra_tf_opdefs,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 08:30:24 UTC 2024 - 4.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/tf_mlir_translate_registration.cc
enable_soft_placement, set_original_tf_func_name}; auto module_or = tensorflow::GraphdefToMlirTranslateFunction( input, input_arrays, input_dtypes, input_shapes, output_arrays, control_output_arrays, options, context); if (!module_or.status().ok()) return nullptr; return std::move(module_or).value(); } static TranslateToMLIRRegistration GraphdefToMlirTranslate(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 7.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tf_to_tfl_flatbuffer.cc
file->getBuffer(), input_arrays, input_dtypes, input_shapes, output_arrays, control_output_arrays, graphdef_conversion_options, context); } return GraphdefToMlirTranslateFunction(file->getBuffer(), input_arrays, input_dtypes, input_shapes, output_arrays, control_output_arrays,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:01:23 UTC 2024 - 23.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tf_tfl_translate.cc
module = tensorflow::LoadFromGraphdefOrMlirSource( input_file_name, input_mlir, use_splatted_constant, custom_opdefs, specs, debug_info_file, input_arrays, input_dtypes, input_shapes, output_arrays, control_output_arrays, &source_mgr, &context); } // If errors occur, the library call in the above already logged the error // message. So we can just return here. if (!module.ok()) return kTrFailure;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:01:23 UTC 2024 - 14K bytes - Viewed (0)