Home
last modified time | relevance | path

Searched defs:resultShape (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
H A Dmhlo_canonicalize_reduction.cc235 Value resultShape = b.create<tensor::FromElementsOp>(loc, resultDims); in runOnOperation() local
H A Dlegalize_mhlo_to_thlo.cc57 const ArrayRef<int64_t> resultShape = resultTy.getShape(); in matchAndRewrite() local
H A Dgroup_reduction_dimensions.cc156 Optional<Value> resultShape; in tryLowerTo1DOr2DReduction() local
H A Drank_specialization.cc362 SmallVector<Value, 1> resultShape; in materializeFinalReshape() local
H A Dlegalize_to_linalg.cc2867 auto resultShape = resultType.getShape(); in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/tosa/lib/Transforms/
H A Dlegalize_mhlo.cc99 auto resultShape = resultType.getShape(); in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/
H A DChloOps.cpp481 SmallVector<int64_t> resultShape; in inferReturnTypeComponents() local
H A DStablehloOps.cpp1249 SmallVector<int64_t> resultShape = to_vector(operandShape); in inferReturnTypeComponents() local
2256 SmallVector<int64_t> resultShape(operandRankedType.getShape().begin(), in inferReturnTypeComponents() local
4328 SmallVector<int64_t> resultShape; in inferReturnTypeComponents() local
4977 SmallVector<int64_t> resultShape; in inferReturnTypes() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/
H A Dchlo_ops.cc508 SmallVector<int64_t> resultShape; in inferReturnTypeComponents() local
H A Dhlo_ops.cc1286 SmallVector<int64_t> resultShape = to_vector(operandShape); in inferReturnTypeComponents() local
2773 SmallVector<int64_t> resultShape(operandRankedType.getShape().begin(), in inferReturnTypeComponents() local
5734 SmallVector<int64_t> resultShape; in inferReturnTypeComponents() local
7411 SmallVector<int64_t> resultShape; in inferReturnTypes() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/gml_st/IR/
H A Dgml_st_ops.cc1604 SmallVector<int64_t> resultShape = llvm::to_vector(llvm::map_range( in inferReturnTypes() local
1713 const SmallVector<int64_t> resultShape = llvm::to_vector(llvm::map_range( in inferReturnTypes() local