- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 384 for mlir_context (0.13 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/convert_tensor_test.cc
namespace { using ::testing::Eq; using ::testing::IsFalse; using ::testing::IsTrue; static void RegisterDialects(mlir::MLIRContext &context) { context.loadDialect<mlir::TF::TensorFlowDialect>(); } TEST(ConvertTypeToTensorTypeTest, UnrankedTensorType) { mlir::MLIRContext context; RegisterDialects(context); mlir::Builder b(&context); PartialTensorShape output_shape =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/utils/dialect_detection_utils_test.cc
#include "tensorflow/compiler/mlir/tf2xla/internal/utils/dialect_detection_utils.h" #include <gtest/gtest.h> #include "mlir/IR/Builders.h" // from @llvm-project #include "mlir/IR/MLIRContext.h" // from @llvm-project #include "mlir/IR/Operation.h" // from @llvm-project #include "mlir/IR/OperationSupport.h" // from @llvm-project #include "mlir/IR/OwningOpRef.h" // from @llvm-project
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Nov 08 20:33:37 UTC 2023 - 2.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/context.h
#include "mlir/IR/MLIRContext.h" // from @llvm-project namespace mlir::quant::stablehlo { // Creates an MLIRContext with the extensions required for quantization are // registered. inline std::unique_ptr<MLIRContext> CreateMlirContextForQuantization() { DialectRegistry registry{}; func::registerAllExtensions(registry); return std::make_unique<MLIRContext>(registry); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jan 11 16:08:42 UTC 2024 - 1.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/tf_mlir_translate.h
#include "absl/strings/string_view.h" #include "absl/types/span.h" #include "mlir/Dialect/Func/IR/FuncOps.h" // from @llvm-project #include "mlir/IR/BuiltinOps.h" // from @llvm-project #include "mlir/IR/MLIRContext.h" // from @llvm-project #include "tensorflow/cc/saved_model/loader.h" #include "tensorflow/compiler/mlir/tensorflow/translate/mlir_import_options.h" namespace tensorflow { using tsl::Status; using tsl::StatusOr;
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/utils/perception_ops_utils_test.cc
ArrayAttr createInt32Array(mlir::Builder* builder, mlir::MLIRContext* context, const SmallVector<int32_t, N>& values) { SmallVector<Attribute, N> ret; for (int32_t value : values) { ret.push_back(builder->getI32IntegerAttr(value)); } return ArrayAttr::get(context, ret); } template <int N> ArrayAttr createInt64Array(mlir::Builder* builder, mlir::MLIRContext* context,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Sep 29 21:02:21 UTC 2022 - 7.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf_mlir_translate_main.cc
if (import_saved_model_object_graph) { mlir::MLIRContext context; auto module_or = tensorflow::SavedModelObjectGraphToMlirImport( input_filename, tags, exported_names, &context); if (!module_or.status().ok()) return 1; module_or.value()->print(output->os()); } else if (import_saved_model_signature_defs) { mlir::MLIRContext context; tensorflow::MLIRImportOptions import_options;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 08 01:19:25 UTC 2023 - 7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.h
const GraphImportConfig& specs, mlir::MLIRContext* context); // Given a Graph, returns a MLIR module containing the graph, expressed with // tf_executor dialect. absl::StatusOr<mlir::OwningOpRef<mlir::ModuleOp>> ConvertGraphToMlir( const Graph& graph, const GraphDebugInfo& debug_info, const FunctionLibraryDefinition& flib_def, const GraphImportConfig& specs, mlir::MLIRContext* context); // [Experimental]
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 6.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/python/mlir_wrapper/basic_classes.cc
#include "mlir/IR/Location.h" // from @llvm-project #include "mlir/IR/MLIRContext.h" // from @llvm-project #include "mlir/IR/Operation.h" // from @llvm-project #include "mlir/IR/Region.h" // from @llvm-project #include "tensorflow/compiler/mlir/python/mlir_wrapper/mlir_wrapper.h" void init_basic_classes(py::module& m) { py::class_<mlir::MLIRContext>(m, "MLIRContext").def(py::init<>()); py::class_<mlir::Location>(m, "Location");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jan 22 04:26:07 UTC 2022 - 2.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/quantize.cc
#include "mlir/Dialect/Quant/QuantOps.h" // from @llvm-project // IWYU pragma: keep #include "mlir/IR/BuiltinAttributes.h" // from @llvm-project #include "mlir/IR/BuiltinOps.h" // from @llvm-project #include "mlir/IR/MLIRContext.h" // from @llvm-project #include "mlir/IR/Operation.h" // from @llvm-project #include "mlir/IR/PatternMatch.h" // from @llvm-project #include "mlir/Pass/Pass.h" // from @llvm-project // IWYU pragma: keep
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 07:08:19 UTC 2024 - 5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/dump_mlir_util_test.cc
#include "mlir/IR/BuiltinOps.h" // from @llvm-project #include "mlir/IR/DialectRegistry.h" // from @llvm-project #include "mlir/IR/Location.h" // from @llvm-project #include "mlir/IR/MLIRContext.h" // from @llvm-project #include "mlir/InitAllDialects.h" // from @llvm-project #include "mlir/InitAllPasses.h" // from @llvm-project #include "mlir/Pass/PassManager.h" // from @llvm-project
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 18 13:40:21 UTC 2024 - 5.8K bytes - Viewed (0)