- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for RefineShapes (0.19 sec)
-
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.h
struct TensorOrResourceShape { TensorShape shape; bool is_resource = false; }; // Refine MLIR types based on new shape information. ABSL_DEPRECATED("Not meant to be used directly and should be a util.") Status RefineShapes(llvm::ArrayRef<TensorOrResourceShape> arg_shapes, mlir::ModuleOp module); // Lower TF to MHLO and insert HLO into the XlaBuilder. xla_params are HLO-level
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 17:24:39 UTC 2024 - 10.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc
// since XLA uses functional control flow. pm.addNestedPass<mlir::func::FuncOp>( mlir::mhlo::createSinkConstantsToControlFlowPass()); } Status RefineShapes(llvm::ArrayRef<TensorOrResourceShape> arg_shapes, mlir::ModuleOp module) { auto producer_or = GetTfGraphProducerVersion(module); if (!producer_or.ok()) return producer_or.status();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 17:24:39 UTC 2024 - 45.3K bytes - Viewed (0)