Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
Dlegalize_shape_computations.cc87 ShapedType operandTy = operand.getType().template cast<ShapedType>(); in matchAndRewrite() local
124 ShapedType operandTy = operand.getType().template cast<ShapedType>(); in matchAndRewrite() local
Dlegalize_mhlo_to_thlo.cc83 auto operandTy = operand.getType().cast<RankedTensorType>(); in matchAndRewrite() local
123 auto operandTy = op.operand().getType().cast<RankedTensorType>(); in matchAndRewrite() local
Dlegalize_gather_to_torch_index_select.cc42 auto operandTy = operand.getType().cast<ShapedType>(); in matchAndRewrite() local
Drestrict_max_rank.cc67 TensorType operandTy = operand.getType().cast<TensorType>(); in matchAndRewrite() local
Dbroadcast_propagation.cc193 auto operandTy = operand.getType().cast<RankedTensorType>(); in findBroadcastIntents() local
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Transforms/
Dsymbolic_shape_optimization.cc369 LogicalResult materializeReshapeAsScalarExpand(RankedTensorType operandTy, in materializeReshapeAsScalarExpand()
387 LogicalResult materializeReshapeAsScalarCollapse(RankedTensorType operandTy, in materializeReshapeAsScalarCollapse()
661 ShapeComponentAnalysis &shapeAnalysis, RankedTensorType operandTy, in materializeReshapeAsExpandAndCollapse()
714 auto operandTy = op.operand().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/
Dchlo_ops.cc491 auto operandTy = operand.getType().dyn_cast<RankedTensorType>(); in inferReturnTypeComponents() local
Dhlo_ops.cc2129 auto operandTy = (*operands.begin()).getType().cast<ShapedType>(); in inferReturnTypes() local
2564 Type operandTy = operand.getType(); in build() local
2574 auto operandTy = getOperand().getType().cast<TensorType>(); in fold() local
4846 static TensorType getReduceResultType(Type operandTy, in getReduceResultType()
5854 auto operandTy = operand.getType().cast<RankedTensorType>(); in reifyReturnTypeShapes() local
5922 auto operandTy = operand.getType().cast<RankedTensorType>(); in matchAndRewrite() local
5974 auto operandTy = operand.getType().cast<RankedTensorType>(); in matchAndRewrite() local
6114 auto operandTy = operand().getType().dyn_cast<RankedTensorType>(); in verify() local
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/
DChloOps.cpp464 auto operandTy = operand.getType().dyn_cast<RankedTensorType>(); in inferReturnTypeComponents() local
DStablehloOps.cpp1806 auto operandTy = (*operands.begin()).getType().cast<ShapedType>(); in inferReturnTypes() local
2165 Type operandTy = operand.getType(); in build() local
3612 static TensorType getReduceResultType(Type operandTy, in getReduceResultType()
4366 auto operandTy = operand.getType().cast<RankedTensorType>(); in reifyReturnTypeShapes() local
4541 auto operandTy = operand().getType().dyn_cast<RankedTensorType>(); in verify() local
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/thlo/IR/
Dthlo_ops.cc405 auto operandTy = operand().getType().cast<RankedTensorType>(); in fuse() local