Searched refs:kernelInputFeatureDimension (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/python/ |
| H A D | MlirHloModule.cc | 243 int64_t kernelInputFeatureDimension, 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 | 68 "int64_t":$kernelInputFeatureDimension,
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/ |
| H A D | legalize_to_linalg.cc | 1998 auto kernelInputFeatureDimension = in matchAndRewrite() local
2181 windowExprs[rhsIndexMapping[kernelInputFeatureDimension]] = in matchAndRewrite()
2334 int64_t kernelInputFeatureDimension = in matchAndRewrite() local
2338 if (filterDims[kernelInputFeatureDimension] * in matchAndRewrite()
2351 if (filterDims[kernelInputFeatureDimension] == 1) { in matchAndRewrite()
2352 reshapedFilterDims[kernelInputFeatureDimension] = in matchAndRewrite()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/ |
| H A D | StablehloAttrs.td | 72 "int64_t":$kernelInputFeatureDimension,
|
| H A D | StablehloOps.cpp | 6078 int64_t kernelInputFeatureDimension = 0; in parseConvolutionDimensionsRaw() local
6096 return parser.parseInteger(kernelInputFeatureDimension); in parseConvolutionDimensionsRaw()
6113 kernelInputFeatureDimension, kernelOutputFeatureDimension, in parseConvolutionDimensionsRaw()
6281 int64_t kernelInputFeatureDimension = parsedDims.second[KIFeature]; 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 | 8904 int64_t kernelInputFeatureDimension = 0; in parseConvolutionDimensionsRaw() local
8922 return parser.parseInteger(kernelInputFeatureDimension); in parseConvolutionDimensionsRaw()
8939 kernelInputFeatureDimension, kernelOutputFeatureDimension, in parseConvolutionDimensionsRaw()
9107 int64_t kernelInputFeatureDimension = parsedDims.second[KIFeature]; in parseConvolutionDimensions() local
9121 kernelInputFeatureDimension, kernelOutputFeatureDimension, in parseConvolutionDimensions()
|