| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/ |
| H A D | ChloOps.cpp | 251 MLIRContext* /*ctx*/, Optional<Location>, ValueRange operands, in inferReturnTypes() argument
262 MLIRContext* /*ctx*/, Optional<Location>, ValueRange operands, in inferReturnTypes() argument
273 MLIRContext* /*ctx*/, Optional<Location>, ValueRange operands, in inferReturnTypes() argument
347 MLIRContext* /*context*/, Optional<Location> location, in inferReturnTypeComponents() argument
381 MLIRContext*, Optional<Location> location, ValueShapeRange operands, in inferReturnTypeComponents() argument
499 MLIRContext*, Optional<Location>, ValueRange, DictionaryAttr attributes, in inferReturnTypes() argument
|
| H A D | StablehloOps.cpp | 757 MLIRContext*, Optional<Location>, ValueRange operands, in inferReturnTypes() argument
942 MLIRContext*, Optional<Location> location, ValueShapeRange operands, in inferReturnTypeComponents() argument
1013 MLIRContext*, Optional<Location>, ValueRange operands, DictionaryAttr, in inferReturnTypes() argument
1193 MLIRContext*, Optional<Location> location, ValueShapeRange operands, in inferReturnTypeComponents() argument
1804 MLIRContext*, Optional<Location>, ValueRange operands, DictionaryAttr, in inferReturnTypes() argument
2221 MLIRContext*, Optional<Location> location, ValueShapeRange operands, in inferReturnTypeComponents() argument
2570 MLIRContext*, Optional<Location> location, ValueShapeRange operands, in inferReturnTypeComponents() argument
2841 MLIRContext*, Optional<Location> /*location*/, ValueShapeRange operands, in inferReturnTypeComponents() argument
2865 MLIRContext*, Optional<Location>, ValueRange operands, DictionaryAttr, in inferReturnTypes() argument
2889 MLIRContext*, Optional<Location>, ValueRange operands, DictionaryAttr, in inferReturnTypes() argument
[all …]
|
| H A D | Base.h | 136 MLIRContext * /*context*/, Optional<Location> location, in inferReturnTypes() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/ |
| H A D | chlo_ops.cc | 278 MLIRContext* /*ctx*/, Optional<Location>, ValueRange operands, in inferReturnTypes() argument
289 MLIRContext* /*ctx*/, Optional<Location>, ValueRange operands, in inferReturnTypes() argument
300 MLIRContext* /*ctx*/, Optional<Location>, ValueRange operands, in inferReturnTypes() argument
374 MLIRContext* /*context*/, Optional<Location> location, in inferReturnTypeComponents() argument
408 MLIRContext*, Optional<Location> location, ValueShapeRange operands, in inferReturnTypeComponents() argument
|
| H A D | hlo_ops.cc | 755 MLIRContext*, Optional<Location>, ValueRange operands, in inferReturnTypes() argument
940 MLIRContext*, Optional<Location> location, ValueShapeRange operands, in inferReturnTypeComponents() argument
1011 MLIRContext*, Optional<Location>, ValueRange operands, DictionaryAttr, in inferReturnTypes() argument
1230 MLIRContext*, Optional<Location> location, ValueShapeRange operands, in inferReturnTypeComponents() argument
2127 MLIRContext*, Optional<Location>, ValueRange operands, DictionaryAttr, in inferReturnTypes() argument
2738 MLIRContext*, Optional<Location> location, ValueShapeRange operands, in inferReturnTypeComponents() argument
3117 MLIRContext*, Optional<Location> location, ValueShapeRange operands, in inferReturnTypeComponents() argument
3554 MLIRContext*, Optional<Location> /*location*/, ValueShapeRange operands, in inferReturnTypeComponents() argument
3578 MLIRContext*, Optional<Location>, ValueRange operands, DictionaryAttr, in inferReturnTypes() argument
3611 MLIRContext*, Optional<Location>, ValueRange operands, DictionaryAttr, in inferReturnTypes() argument
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Transforms/ |
| H A D | rewriters.h | 25 class MLIRContext; variable
|
| H A D | passes.h | 26 class MLIRContext; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/gml_st/transforms/ |
| H A D | rewriters.h | 25 class MLIRContext; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/ |
| H A D | xla_mlir_translate.h | 28 class MLIRContext; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| H A D | rewriters.h | 27 class MLIRContext; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/ |
| H A D | lower_complex.cc | 68 void mlir::mhlo::populateComplexLoweringPatterns(MLIRContext* /*context*/, in populateComplexLoweringPatterns() argument
|
| H A D | materialize_broadcasts.cc | 73 void setupMaterializeBroadcastsLegality(MLIRContext * /*context*/, in setupMaterializeBroadcastsLegality() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| H A D | tf_ops_layout_helper.h | 30 class MLIRContext; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| H A D | tfl_ops.cc | 939 MLIRContext *, Optional<Location> location, ValueRange operands, in inferReturnTypes() argument
1204 MLIRContext *, Optional<Location> location, ValueRange operands, in inferReturnTypes() argument
2607 MLIRContext *, Optional<Location>, ValueRange operands, DictionaryAttr attr, in inferReturnTypes() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/gml_st/IR/ |
| H A D | gml_st_ops.cc | 140 MLIRContext *, Optional<Location>, ValueRange operands, in inferReturnTypes() argument
|