Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dmhlo_mark_shape_calc.cc196 auto operand_ty = in MarkRegardAsShapeCalcOps() local
207 auto operand_ty = in MarkRegardAsShapeCalcOps() local
Dlegalize_gather_to_torch_index_select.cc41 auto operand_ty = operand.getType().cast<ShapedType>(); in matchAndRewrite() local
Dbroadcast_propagation.cc375 auto operand_ty = operand.getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local
Dlegalize_to_linalg.cc2582 auto operand_ty = adaptor.operand().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/
Dhlo_ops.cc716 auto operand_ty = (*operands.begin()).getType().cast<ShapedType>(); in inferReturnTypes() local
748 Type operand_ty = operand.getType(); in build() local
758 auto operand_ty = getOperand().getType().cast<TensorType>(); in fold() local
2314 static TensorType GetReduceResultType(Type operand_ty, in GetReduceResultType()
2873 auto operand_ty = op.operand().getType().dyn_cast<RankedTensorType>(); in Verify() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc1642 auto operand_ty = val.getType().dyn_cast<ShapedType>(); in VerifyStridedSliceBase() local
3252 TensorType operand_ty = op.input().getType().cast<TensorType>(); in inferReturnTypes() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc3267 auto operand_ty = operand.getType().cast<TensorType>(); in matchAndRewrite() local