Home
last modified time | relevance | path

Searched defs:shapedTy (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
Dlegalize_shape_computations.cc60 auto shapedTy = operand.getType().template cast<ShapedType>(); in opIsShapeComputation() local
172 auto shapedTy = operand.getType().template cast<ShapedType>(); in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/
DBase.cpp114 auto shapedTy = operand.getType().dyn_cast<ShapedType>(); in deriveShapeFromOperand() local
DStablehloOps.cpp2959 auto shapedTy = type.cast<ShapedType>(); in inferReturnTypes() local
4953 auto shapedTy = type.dyn_cast<ShapedType>(); in inferReturnTypes() local
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/
Dhlo_ops.cc3785 auto shapedTy = type.cast<ShapedType>(); in inferReturnTypes() local
7387 auto shapedTy = type.dyn_cast<ShapedType>(); in inferReturnTypes() local
9417 auto shapedTy = operand.getType().dyn_cast<ShapedType>(); in deriveShapeFromOperand() local