Searched defs:input_ty (Results 1 – 8 of 8) sorted by relevance
341 auto input_ty = input.getType().cast<ShapedType>(); in matchAndRewrite() local760 Type input_ty = input.getType(); in matchAndRewrite() local1034 auto input_ty = input.getType().cast<ShapedType>(); in matchAndRewrite() local1302 auto input_ty = input.getType().cast<ShapedType>(); in matchAndRewrite() local
1612 if (auto input_ty = func_type.getInput(i).dyn_cast<RankedTensorType>()) { in InferShapeForFunction() local
986 auto input_ty = input.getType().dyn_cast<RankedTensorType>(); in CanBeTranslatedToDynamicSlice() local1020 auto input_ty = input.getType().dyn_cast<RankedTensorType>(); in TFSliceSizes2HLOSliceSizes() local1171 auto input_ty = op.input().getType().template dyn_cast<RankedTensorType>(); in matchAndRewrite() local1729 auto input_ty = op.input().getType().template cast<ShapedType>(); in matchAndRewrite() local2528 auto input_ty = op.input().getType().template dyn_cast<RankedTensorType>(); in matchAndRewrite() local3103 RankedTensorType input_ty, in rewriteWithConstantBegin()3162 RankedTensorType input_ty, in rewriteWithUnknownBegin()3302 auto input_ty = op.input().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local3624 auto input_ty = op.input().getType().template dyn_cast<RankedTensorType>(); in matchAndRewrite() local3982 auto input_ty = op.input().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local[all …]
1292 auto input_ty = op.x().getType().template dyn_cast<RankedTensorType>(); in Verify() local1575 const TensorType input_ty = input.getType().template cast<TensorType>(); in inferConvReturnTypes() local1703 auto input_ty = input().getType().dyn_cast<TensorType>(); in GetOptimalLayout() local1786 auto input_ty = input().getType().dyn_cast<TensorType>(); in GetOptimalLayout() local1898 auto input_ty = op.x().getType().dyn_cast<RankedTensorType>(); in Verify() local2130 auto input_ty = input.getType().dyn_cast<RankedTensorType>(); in InferExpandDimsOpType() local
889 static Attribute ConvertShapeToAttr(Type input_ty, int out_width) { in ConvertShapeToAttr()1084 auto input_ty = op.input().getType().dyn_cast<RankedTensorType>(); in Verify() local1550 auto input_ty = input().getType().template dyn_cast<RankedTensorType>(); in fold() local1870 auto input_ty = this->input().getType().dyn_cast<RankedTensorType>(); in GetSlicedBoundRanges() local2558 auto input_ty = nms_op.getType().template cast<ShapedType>(); in matchAndRewrite() local
322 def quantizer_flags(self, input_ty=None, output_ty=None): argument347 def flags_modify_model_io_type(self, input_ty=None, output_ty=None): argument
1352 auto input_ty = input.getType().cast<TensorType>(); in GetReshapeOutputType() local1453 auto input_ty = op.input().getType().cast<TensorType>(); in Verify() local
2550 ShapedType input_ty = input.getType().cast<ShapedType>(); in matchAndRewrite() local