- Sort Score
- Result 10 results
- Languages All
Results 11 - 15 of 15 for registry_ (0.38 sec)
-
RELEASE.md
### TPU Enhancements * Adds 3D mesh support in TPU configurations ops. * Added TPU code for `FTRL` with `multiply_linear_by_lr`. * Silently adds a new file system registry at `gstpu`. * Support `restartType` in cloud tpu client. * Depend on a specific version of google-api-python-client. * Fixes apiclient import. ### Tracing and Debugging
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 730.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/lite/quantize_weights.cc
absl::flat_hash_set<std::string> denylisted_mlir_op_names; for (auto& entry : denylisted_ops) { denylisted_mlir_op_names.insert(TfLiteToMlir(entry)); } DialectRegistry registry; MLIRContext context(registry); StatusScopedDiagnosticHandler statusHandler(&context, /*propagate=*/true); // Import input_model to a MLIR module
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 23:15:24 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_tf_graph.cc
xla::CompileOnlyClient* client, XlaCompiler::CompilationResult* compilation_result) { mlir::DialectRegistry registry; mlir::RegisterAllTensorFlowDialects(registry); mlir::mhlo::registerAllMhloDialects(registry); mlir::MLIRContext context(registry); mlir::OwningOpRef<mlir::ModuleOp> mlir_module; TF_RETURN_IF_ERROR(DeserializeMlirModule(mlir_computation.mlir_module,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 14K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_graph_optimization_pass.cc
class GraphOptPass : public mlir::PassWrapper<GraphOptPass, mlir::OperationPass<mlir::ModuleOp>> { void getDependentDialects(mlir::DialectRegistry& registry) const override { mlir::RegisterAllTensorFlowDialects(registry); } public: MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(GraphOptPass) explicit GraphOptPass(std::vector<tensorflow::GraphOptimizationPass*> passes)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 7.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
// It is a terrible idea to have local MLIR contexts so we need to // register extensions here, again. mlir::DialectRegistry registry; registry.insert<mlir::func::FuncDialect>(); mlir::func::registerAllExtensions(registry); xla_call_module_context_.appendDialectRegistry(registry); auto l = tensorflow::XlaCallModuleLoader::Create( &xla_call_module_context_, op.getVersion(), op.getModule().str(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0)