Home
last modified time | relevance | path

Searched defs:inputBatchDimension (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/python/
H A DMlirHloModule.cc248 MlirContext ctx) { in PYBIND11_MODULE()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/CAPI/
H A DAttributes.cc247 MlirContext ctx, int64_t inputBatchDimension, int64_t inputFeatureDimension, in mlirMhloConvDimensionNumbersGet()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/
H A DStablehloOps.cpp6075 int64_t inputBatchDimension = 0; in parseConvolutionDimensionsRaw() local
6274 int64_t inputBatchDimension = parsedDims.second[IOBatch]; in parseConvolutionDimensions() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
H A Dlegalize_to_linalg.cc1994 auto inputBatchDimension = dimensionNumbers.getInputBatchDimension(); in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/
H A Dhlo_ops.cc8901 int64_t inputBatchDimension = 0; in parseConvolutionDimensionsRaw() local
9100 int64_t inputBatchDimension = parsedDims.second[IOBatch]; in parseConvolutionDimensions() local