Searched refs:ShapedTypeComponents (Results 1 – 11 of 11) sorted by relevance
| /external/llvm-project/mlir/include/mlir/Interfaces/ |
| D | InferTypeOpInterface.h | 34 class ShapedTypeComponents {
40 ShapedTypeComponents() : ranked(false), elementType(nullptr), attr(nullptr){}; in ShapedTypeComponents() function
41 ShapedTypeComponents(Type elementType) in ShapedTypeComponents() function
45 ShapedTypeComponents(Arg &&arg, Type elementType = nullptr,
49 ShapedTypeComponents(ArrayRef<int64_t> vec, Type elementType = nullptr,
87 SmallVectorImpl<ShapedTypeComponents> &retComponents)>
|
| D | InferTypeOpInterface.td | 100 "SmallVectorImpl<ShapedTypeComponents>&":
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | shape_inference_utils.cc | 163 ShapedTypeComponents CreateShapedTypeComponents(InferenceContext& context, in CreateShapedTypeComponents()
168 return ShapedTypeComponents(shape.getValue(), element_type); in CreateShapedTypeComponents()
169 return ShapedTypeComponents(element_type); in CreateShapedTypeComponents()
179 SmallVectorImpl<ShapedTypeComponents>& inferred_return_shapes) { in InferReturnTypeComponentsForTFOp()
342 SmallVectorImpl<ShapedTypeComponents>& inferred_return_shapes) { in InferReturnTypeComponentsForTFOp()
358 ShapedTypeComponents(shaped_type.getShape(), in InferReturnTypeComponentsForTFOp()
362 ShapedTypeComponents(shaped_type.getElementType()); in InferReturnTypeComponentsForTFOp()
|
| D | shape_inference_utils.h | 59 SmallVectorImpl<ShapedTypeComponents>& inferred_return_shapes);
71 SmallVectorImpl<ShapedTypeComponents>& inferred_return_shapes);
|
| /external/llvm-project/mlir/lib/Interfaces/ |
| D | InferTypeOpInterface.cpp | 28 SmallVectorImpl<ShapedTypeComponents> &retComponents)> in inferReturnTensorTypes()
33 SmallVector<ShapedTypeComponents, 2> retComponents; in inferReturnTensorTypes()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | chlo_ops.cc | 130 SmallVectorImpl<ShapedTypeComponents>& inferedReturnShapes) { in InferBroadcastBinaryOpReturnTypeComponents()
197 SmallVectorImpl<ShapedTypeComponents>& inferedReturnShapes) { in inferReturnTypeComponents()
231 SmallVectorImpl<ShapedTypeComponents>& inferedReturnShapes) { in inferReturnTypeComponents()
291 SmallVectorImpl<ShapedTypeComponents>& inferedReturnShapes) { \
343 SmallVectorImpl<ShapedTypeComponents>& inferedReturnShapes) { in inferReturnTypeComponents()
|
| D | hlo_ops.cc | 1871 llvm::SmallVectorImpl<mlir::ShapedTypeComponents>&) { in inferReturnTypeComponents() argument
2845 llvm::SmallVectorImpl<mlir::ShapedTypeComponents>&) { in inferReturnTypeComponents() argument
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | test_infer_shaped_type_pass.cc | 38 SmallVector<ShapedTypeComponents, 4> components; in matchAndRewrite()
|
| /external/llvm-project/mlir/test/lib/Dialect/Test/ |
| D | TestDialect.cpp | 694 SmallVectorImpl<ShapedTypeComponents> &inferredReturnShapes) { in inferReturnTypeComponents()
704 inferredReturnShapes.push_back(ShapedTypeComponents({dim}, type)); in inferReturnTypeComponents()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | shape_inference.cc | 1107 llvm::SmallVector<ShapedTypeComponents, 4> inferred_return_shapes; in InferShapeForSingleOperation()
1121 ShapedTypeComponents inferred = std::get<1>(result); in InferShapeForSingleOperation()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | hlo_ops.td | 132 SmallVectorImpl<ShapedTypeComponents>& inferedReturnShapes) {
286 SmallVectorImpl<ShapedTypeComponents>& inferedReturnShapes) {
|