Searched defs:input_ty (Results 1 – 10 of 10) sorted by relevance
73 PrimitiveType input_ty = instr->operand(0)->shape().element_type(); in HeuristicLayoutAssignment() local
59 PrimitiveType input_ty = instr->operand(0)->shape().element_type(); in GetRelevantConvs() local
385 auto input_ty = input.getType().cast<ShapedType>(); in matchAndRewrite() local801 Type input_ty = input.getType(); in matchAndRewrite() local1072 auto input_ty = input.getType().cast<ShapedType>(); in matchAndRewrite() local1338 auto input_ty = input.getType().cast<ShapedType>(); in matchAndRewrite() local1555 auto input_ty = tf_roll_op.input().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local
1840 if (auto input_ty = func_type.getInput(i).dyn_cast<RankedTensorType>()) { in InferShapeForFunction() local
101 RankedTensorType input_ty) { in GetFeatureDimension()944 auto input_ty = input.getType().dyn_cast<RankedTensorType>(); in CanBeTranslatedToDynamicSlice() local978 auto input_ty = input.getType().dyn_cast<RankedTensorType>(); in TFSliceSizes2HLOSliceSizes() local1310 auto input_ty = op.input().getType().template dyn_cast<RankedTensorType>(); in matchAndRewriteDynamicConv() local1469 auto input_ty = op.input().getType().template dyn_cast<RankedTensorType>(); in matchAndRewrite() local1842 auto input_ty = op.input().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local2336 auto input_ty = op.input().getType().template cast<ShapedType>(); in matchAndRewrite() local3142 auto input_ty = op.input().getType().template dyn_cast<RankedTensorType>(); in matchAndRewrite() local3297 auto input_ty = input.getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local3787 RankedTensorType input_ty, in rewriteWithConstantBegin()[all …]
1310 auto input_ty = op.x().getType().template dyn_cast<RankedTensorType>(); in Verify() local1598 const TensorType input_ty = input.getType().template cast<TensorType>(); in inferConvReturnTypes() local1726 auto input_ty = input().getType().dyn_cast<TensorType>(); in GetOptimalLayout() local1809 auto input_ty = input().getType().dyn_cast<TensorType>(); in GetOptimalLayout() local1921 auto input_ty = op.x().getType().dyn_cast<RankedTensorType>(); in Verify() local2331 auto input_ty = input.getType().dyn_cast<RankedTensorType>(); in InferExpandDimsOpType() local2364 auto input_ty = operands[0].getType().dyn_cast<TensorType>(); in inferReturnTypes() local
326 auto input_ty = pack_op.getOperand(0).getType().cast<ShapedType>(); in matchAndRewrite() local1015 static Attribute ConvertShapeToAttr(Type input_ty, int out_width) { in ConvertShapeToAttr()1210 auto input_ty = op.input().getType().dyn_cast<RankedTensorType>(); in Verify() local1605 auto input_ty = input().getType().template dyn_cast<RankedTensorType>(); in fold() local1925 auto input_ty = this->input().getType().dyn_cast<RankedTensorType>(); in GetSlicedBoundRanges() local2623 auto input_ty = nms_op.getType().template cast<ShapedType>(); in matchAndRewrite() local
253 auto input_ty = input.getType().cast<ShapedType>(); in CanOptimizeIdentitySliceOp() local1267 auto input_ty = input.getType().dyn_cast<ShapedType>(); in matchAndRewrite() local
1070 const RankedTensorType input_ty = in inferReturnTypes() local1613 auto input_ty = input.getType().cast<TensorType>(); in GetReshapeOutputType() local1714 auto input_ty = op.input().getType().cast<TensorType>(); in Verify() local1966 auto input_ty = input().getType().dyn_cast<RankedTensorType>(); in fold() local
1883 auto input_ty = input.getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local3495 ShapedType input_ty = input.getType().cast<ShapedType>(); in matchAndRewrite() local