Home
last modified time | relevance | path

Searched refs:splitDimension (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/
H A DStablehloOps.cpp2234 int64_t splitDimension = static_cast<int64_t>(adaptor.split_dimension()); in inferReturnTypeComponents() local
2236 if (splitDimension >= inputRank || splitDimension < 0) { in inferReturnTypeComponents()
2238 splitDimension, in inferReturnTypeComponents()
2250 auto splitDimSize = operandRankedType.getDimSize(splitDimension); in inferReturnTypeComponents()
2258 resultShape[splitDimension] /= splitCount; in inferReturnTypeComponents()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/
H A Dhlo_ops.cc2751 int64_t splitDimension = static_cast<int64_t>(adaptor.split_dimension()); in inferReturnTypeComponents() local
2753 if (splitDimension >= inputRank || splitDimension < 0) { in inferReturnTypeComponents()
2755 splitDimension, in inferReturnTypeComponents()
2767 auto splitDimSize = operandRankedType.getDimSize(splitDimension); in inferReturnTypeComponents()
2775 resultShape[splitDimension] /= splitCount; in inferReturnTypeComponents()
/aosp_15_r20/prebuilts/bundletool/
HDbundletool-all-20230807.jarMETA-INF/ META-INF/MANIFEST.MF com/ com/android/ com/ ...