- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 15 for op_or_arg_name_mapper (0.29 sec)
-
tensorflow/compiler/mlir/op_or_arg_name_mapper.cc
See the License for the specific language governing permissions and limitations under the License. ==============================================================================*/ #include "tensorflow/compiler/mlir/op_or_arg_name_mapper.h" #include <string> #include "absl/strings/string_view.h" #include "llvm/ADT/APInt.h" #include "llvm/ADT/STLExtras.h" #include "llvm/ADT/SmallString.h" #include "llvm/ADT/SmallVector.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_export.h
#include <optional> #include <string> #include <unordered_set> #include "mlir/Dialect/Func/IR/FuncOps.h" // from @llvm-project #include "mlir/IR/BuiltinOps.h" // from @llvm-project #include "tensorflow/compiler/mlir/op_or_arg_name_mapper.h" #include "tensorflow/lite/toco/toco_flags.pb.h" namespace tflite { // Options for exporting to Flatbuffer. struct FlatbufferExportOptions { // TocoFlags proto. The following fields are migrated.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 09 02:51:43 UTC 2023 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/BUILD
# To reference all tablegen files here when checking for updates to them. filegroup( name = "td_files", srcs = glob(["**/*.td"]), ) cc_library( name = "op_or_arg_name_mapper", srcs = ["op_or_arg_name_mapper.cc"], hdrs = ["op_or_arg_name_mapper.h"], deps = [ "//tensorflow/compiler/mlir/utils:name_utils", "@com_google_absl//absl/strings", "@llvm-project//llvm:Support",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 9.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_translate.cc
ModuleOp module, llvm::raw_ostream& output) { std::string serialized_flatbuffer; std::unique_ptr<tensorflow::OpOrArgNameMapper> op_or_arg_name_mapper; if (strip_debug_info) { op_or_arg_name_mapper = std::make_unique<tensorflow::OpOrArgStripNameMapper>(); } else { op_or_arg_name_mapper = std::make_unique<tensorflow::OpOrArgLocNameMapper>(); } tflite::FlatbufferExportOptions options;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 14 19:15:40 UTC 2024 - 9.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tf_to_tfl_flatbuffer.cc
} // Write MLIR Stablehlo dialect into FlatBuffer OpOrArgLocNameMapper op_or_arg_name_mapper; tflite::FlatbufferExportOptions options; options.toco_flags = toco_flags; options.saved_model_tags = saved_model_tags; options.op_or_arg_name_mapper = &op_or_arg_name_mapper; options.metadata[tflite::kModelUseStablehloTensorKey] = "true";
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/tf2xla/transforms/tf2xla_rewriter.h
#include "mlir/IR/Types.h" // from @llvm-project #include "mlir/IR/Value.h" // from @llvm-project #include "mlir/Support/LogicalResult.h" // from @llvm-project #include "tensorflow/compiler/mlir/op_or_arg_name_mapper.h" #include "tensorflow/compiler/tf2xla/xla_context.h" #include "tensorflow/compiler/tf2xla/xla_expression.h" #include "xla/client/xla_builder.h" #include "xla/client/xla_computation.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:16:07 UTC 2024 - 5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/prepare_tpu_computation_for_tf_export.cc
#include "mlir/Support/LLVM.h" // from @llvm-project #include "mlir/Support/LogicalResult.h" // from @llvm-project #include "mlir/Transforms/GreedyPatternRewriteDriver.h" // from @llvm-project #include "tensorflow/compiler/mlir/op_or_arg_name_mapper.h" #include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.h" #include "tensorflow/compiler/mlir/tensorflow/ir/tf_remaining_ops.h" #include "tensorflow/compiler/mlir/tensorflow/utils/serialize_mlir_module_utils.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/while_loop_outline.cc
#include "mlir/Transforms/RegionUtils.h" // from @llvm-project #include "tensorflow/compiler/mlir/lite/ir/tfl_ops.h" #include "tensorflow/compiler/mlir/lite/transforms/passes.h" #include "tensorflow/compiler/mlir/op_or_arg_name_mapper.h" #include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.h" namespace mlir { namespace TFL { namespace { #define GEN_PASS_DEF_WHILEOUTLINEPASS #include "tensorflow/compiler/mlir/lite/transforms/passes.h.inc"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_with_tf2xla.cc
#include "mlir/IR/Value.h" // from @llvm-project #include "mlir/Support/LLVM.h" // from @llvm-project #include "mlir/Support/LogicalResult.h" // from @llvm-project #include "tensorflow/compiler/mlir/op_or_arg_name_mapper.h" #include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.h" #include "tensorflow/compiler/mlir/tensorflow/ir/tpu_embedding_ops_registry.h" #include "tensorflow/compiler/mlir/tf2xla/transforms/legalization_op_config.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 9.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
const std::unordered_set<std::string>& tags, OpOrArgNameMapper* op_or_arg_name_mapper, const std::map<std::string, std::string>& metadata, bool serialize_stablehlo_ops, std::optional<size_t> custom_option_alignment) { OpOrArgLocNameMapper default_op_or_arg_name_mapper; if (!op_or_arg_name_mapper) op_or_arg_name_mapper = &default_op_or_arg_name_mapper;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0)