Searched refs:kernelOutputFeatureDimension (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/ |
| H A D | legalize_to_linalg.cc | 2000 auto kernelOutputFeatureDimension = in matchAndRewrite() local
2109 windowExprs.insert(windowExprs.begin() + kernelOutputFeatureDimension, in matchAndRewrite()
2113 reshapeShapeVector(prevDimsRef, newShape, kernelOutputFeatureDimension, in matchAndRewrite()
2116 kernelOutputFeatureDimension); in matchAndRewrite()
2152 windowExprs.insert(windowExprs.begin() + kernelOutputFeatureDimension, in matchAndRewrite()
2156 reshapeShapeVector(prevDimsRef, newShape, kernelOutputFeatureDimension, in matchAndRewrite()
2159 kernelOutputFeatureDimension); in matchAndRewrite()
2190 windowExprs[rhsIndexMapping[kernelOutputFeatureDimension]] = in matchAndRewrite()
2336 int64_t kernelOutputFeatureDimension = in matchAndRewrite() local
2339 filterDims[kernelOutputFeatureDimension] != in matchAndRewrite()
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/python/ |
| H A D | MlirHloModule.cc | 244 int64_t kernelOutputFeatureDimension, in PYBIND11_MODULE()
252 kernelInputFeatureDimension, kernelOutputFeatureDimension, in PYBIND11_MODULE()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/CAPI/ |
| H A D | Attributes.cc | 249 int64_t kernelInputFeatureDimension, int64_t kernelOutputFeatureDimension, in mlirMhloConvDimensionNumbersGet() argument
256 kernelInputFeatureDimension, kernelOutputFeatureDimension, in mlirMhloConvDimensionNumbersGet()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo-c/ |
| H A D | Attributes.h | 132 int64_t kernelInputFeatureDimension, int64_t kernelOutputFeatureDimension,
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| H A D | hlo_ops_base_attrs.td | 69 "int64_t":$kernelOutputFeatureDimension,
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/ |
| H A D | StablehloAttrs.td | 73 "int64_t":$kernelOutputFeatureDimension,
|
| H A D | StablehloOps.cpp | 6079 int64_t kernelOutputFeatureDimension = 0; in parseConvolutionDimensionsRaw() local
6099 return parser.parseInteger(kernelOutputFeatureDimension); in parseConvolutionDimensionsRaw()
6113 kernelInputFeatureDimension, kernelOutputFeatureDimension, in parseConvolutionDimensionsRaw()
6282 int64_t kernelOutputFeatureDimension = parsedDims.second[KOFeature]; in parseConvolutionDimensions() local
6295 kernelInputFeatureDimension, kernelOutputFeatureDimension, in parseConvolutionDimensions()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/ |
| H A D | hlo_ops.cc | 8905 int64_t kernelOutputFeatureDimension = 0; in parseConvolutionDimensionsRaw() local
8925 return parser.parseInteger(kernelOutputFeatureDimension); in parseConvolutionDimensionsRaw()
8939 kernelInputFeatureDimension, kernelOutputFeatureDimension, in parseConvolutionDimensionsRaw()
9108 int64_t kernelOutputFeatureDimension = parsedDims.second[KOFeature]; in parseConvolutionDimensions() local
9121 kernelInputFeatureDimension, kernelOutputFeatureDimension, in parseConvolutionDimensions()
|