- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for SetTPUInfeedLayout (0.39 sec)
-
tensorflow/compiler/mlir/tf2xla/api/v1/compile_tf_graph.cc
mlir::OwningOpRef<mlir::ModuleOp> mlir_module; TF_RETURN_IF_ERROR(DeserializeMlirModule(mlir_computation.mlir_module, &context, &mlir_module)); if (!mlir::SetTPUInfeedLayout(mlir_module)) return errors::Internal("Failed to set layouts attribute"); if (VLOG_IS_ON(2)) { tensorflow::DumpMlirOpToFile("legalize_with_old_bridge", mlir_module.get()); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 14K bytes - Viewed (0)