Searched defs:inferred_return_shapes (Results 1 – 4 of 4) sorted by relevance
31 SmallVectorImpl<ShapedTypeComponents>& inferred_return_shapes) { in InferReturnTypeComponentsForTFOp()
204 SmallVectorImpl<ShapedTypeComponents>& inferred_return_shapes) { in InferReturnTypeComponentsForTFOp()398 SmallVectorImpl<ShapedTypeComponents>& inferred_return_shapes) { in InferReturnTypeComponentsForTFOp()
130 llvm::SmallVectorImpl<mlir::ShapedTypeComponents>& inferred_return_shapes) { in InferShapeOfTFOpWithCustomOperandConstantFn()
2146 llvm::SmallVector<ShapedTypeComponents, 4> inferred_return_shapes; in InferShapeForSingleOperation() local