- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for getfp (0.03 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
// op. The MapDataset op always has N+1 inputs. // TODO(jpienaar): Avoid this lookup. auto module = op->getParentOfType<ModuleOp>(); auto f = module.lookupSymbol<func::FuncOp>(op.getF()); // Skip if function is not found or more than one caller. if (!f || !llvm::hasSingleElement(GetCallers(f))) return false; return InferShapeForDatasetOpCommon(op, f, max_iterations); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0)