Home
last modified time | relevance | path

Searched defs:argTy (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
H A Dgroup_reduction_dimensions.cc39 ReduceOp op, RankedTensorType argTy, Value arg, in tryLowerToCollapseShape()
93 void groupDimensions(RankedTensorType argTy, in groupDimensions()
131 ReduceOp op, RankedTensorType argTy, Value arg, in tryLowerTo1DOr2DReduction()
269 auto argTy = arg.getType().cast<RankedTensorType>(); in matchAndRewrite() local
H A Dlegalize_trigonometric_to_approximation.cc57 Type argTy = arg.getType(); in matchAndRewrite() local
/aosp_15_r20/external/swiftshader/third_party/llvm-10.0/llvm/lib/Target/AMDGPU/
H A DAMDGPULibFunc.cpp976 const Type* argTy = static_cast<const Type*>(*PI); in getOrInsertFunction() local
/aosp_15_r20/external/swiftshader/third_party/llvm-16.0/llvm/lib/Target/AMDGPU/
H A DAMDGPULibFunc.cpp982 const Type* argTy = static_cast<const Type*>(*PI); in getOrInsertFunction() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/gml_st/IR/
H A Dgml_st_ops.cc1592 Type argTy = adaptor.superset().getType(); in inferReturnTypes() local
1702 const Type argTy = adaptor.superset().getType(); in inferReturnTypes() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/
H A DStablehloOps.cpp2903 auto argTy = operands.front().getType().cast<TensorType>(); in inferReturnTypes() local
5094 auto argTy = operands.front().getType().cast<TensorType>(); in inferReturnTypeComponents() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/
H A Dhlo_ops.cc3644 auto argTy = operands.front().getType().cast<TensorType>(); in inferReturnTypes() local
7537 auto argTy = operands.front().getType().cast<TensorType>(); in inferReturnTypeComponents() local