- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 295 for ModuleOp (0.29 sec)
-
tensorflow/compiler/mlir/lite/transforms/passes.td
let summary = "Insert CallOnce op when tf_saved_model's session initializer is give."; let constructor = "CreateInsertCallOnceOpFromSessionInitializerPass()"; let dependentDialects = ["TFL::TensorFlowLiteDialect"]; } def LegalizeHashTablesPass : Pass<"tfl-legalize-hashtables-tf", "mlir::ModuleOp"> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 20:30:06 UTC 2024 - 22.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/lift_hashtable_ops_as_args.cc
namespace mlir { namespace quant { namespace { constexpr StringRef kSharedNameAttr = "shared_name"; class LiftHashTableOpsAsArgsPass : public PassWrapper<LiftHashTableOpsAsArgsPass, OperationPass<ModuleOp>> { public: MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(LiftHashTableOpsAsArgsPass) explicit LiftHashTableOpsAsArgsPass() = default; StringRef getArgument() const final {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 8.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/integration/tfr_decompose_ctx.cc
// Load the TFR functions in a mlir::ModuleOp auto memory_buffer = llvm::MemoryBuffer::getMemBuffer( llvm::StringRef(tfr_raw_text.data(), tfr_raw_text.size())); llvm::SourceMgr source_mgr; source_mgr.AddNewSourceBuffer(std::move(memory_buffer), llvm::SMLoc()); mlir::OwningOpRef<mlir::ModuleOp> module = mlir::parseSourceFile<mlir::ModuleOp>(source_mgr, mlir_ctx); // The MLIRContext owns the module
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 29 02:34:43 UTC 2024 - 9.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/saved_model_export.h
See the License for the specific language governing permissions and limitations under the License. ==============================================================================*/ // Functionalities for exporting MLIR ModuleOp to TensorFlow SavedModel. #ifndef TENSORFLOW_COMPILER_MLIR_QUANTIZATION_STABLEHLO_CC_SAVED_MODEL_EXPORT_H_ #define TENSORFLOW_COMPILER_MLIR_QUANTIZATION_STABLEHLO_CC_SAVED_MODEL_EXPORT_H_ #include <optional>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 20 11:11:25 UTC 2024 - 6.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/cluster_util_test.cc
namespace { constexpr StringRef kTestClusterName = "tpu0"; absl::StatusOr<OwningOpRef<ModuleOp>> GetMlirModuleFromString( StringRef string, MLIRContext* context) { DialectRegistry mlir_registry; RegisterAllTensorFlowDialects(mlir_registry); context->appendDialectRegistry(mlir_registry); OwningOpRef<ModuleOp> mlir_module; auto status = tensorflow::DeserializeMlirModule(string, context, &mlir_module);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 7.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/test_utils.cc
using ::mlir::MLIRContext; using ::mlir::ModuleOp; using ::mlir::OwningOpRef; using ::tsl::StatusOr; absl::StatusOr<OwningOpRef<ModuleOp>> GetMlirModuleFromString( absl::string_view module_string, MLIRContext* context) { DialectRegistry mlir_registry; RegisterCommonToolingDialects(mlir_registry); context->appendDialectRegistry(mlir_registry); OwningOpRef<ModuleOp> mlir_module; auto status =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:16:07 UTC 2024 - 1.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tf_to_tfl_flatbuffer.h
// 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/tf2xla/api/v1/tf_dialect_to_executor_test.cc
#include "tsl/lib/core/status_test_util.h" #include "tsl/platform/status.h" namespace tensorflow { namespace tf2xla { namespace v1 { namespace { using mlir::ModuleOp; using mlir::DialectRegistry; using mlir::MLIRContext; using mlir::ModuleOp; using mlir::OwningOpRef; using tensorflow::monitoring::testing::CellReader; static constexpr char kCompilationStreamz[] =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 08:08:57 UTC 2024 - 3.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/lift_as_function_call_test.cc
return %0 : tensor<1x3xf32> } } )mlir"; TEST_F(LiftAsFunctionCallTest, LiftedFunctionSucceeds) { OwningOpRef<ModuleOp> module_op = ParseModuleOpString(kModuleLifted); ASSERT_TRUE(module_op); auto composite_dot_general_fn = module_op->lookupSymbol<func::FuncOp>("composite_dot_general_fn_1");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 26.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/translate_utils.h
// graph VersionDef proto. void PopulateTfVersions(mlir::ModuleOp module, const VersionDef& versions); // Extracts TensorFlow GraphDef version information from the given module. // Returns failure if version attribute is missing or any of the sub attributes // are invalid. mlir::LogicalResult ExtractTfVersions(mlir::ModuleOp module, VersionDef* versions);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 1.9K bytes - Viewed (0)