Searched defs:input_ty (Results 1 – 11 of 11) sorted by relevance
78 RankedTensorType input_ty) { in GetFeatureDimension()870 auto input_ty = input.getType().dyn_cast<RankedTensorType>(); in CanBeTranslatedToDynamicSlice() local904 auto input_ty = input.getType().dyn_cast<RankedTensorType>(); in TFSliceSizes2HLOSliceSizes() local1128 auto input_ty = op.input().getType().template dyn_cast<RankedTensorType>(); in matchAndRewriteDynamicConv() local1291 auto input_ty = op.input().getType().template dyn_cast<RankedTensorType>(); in matchAndRewrite() local1662 auto input_ty = op.input().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local2127 auto input_ty = op.input().getType().template cast<ShapedType>(); in matchAndRewrite() local2926 auto input_ty = op.input().getType().template dyn_cast<RankedTensorType>(); in matchAndRewrite() local3084 auto input_ty = input.getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local3595 RankedTensorType input_ty, in rewriteWithConstantBegin()[all …]
343 auto input_ty = pack_op.getOperand(0).getType().cast<ShapedType>(); in matchAndRewrite() local1063 static Attribute ConvertShapeToAttr(Type input_ty, int out_width) { in ConvertShapeToAttr()1261 auto input_ty = op.input().getType().dyn_cast<RankedTensorType>(); in verify() local1688 auto input_ty = input().getType().template dyn_cast<RankedTensorType>(); in fold() local2010 auto input_ty = this->input().getType().dyn_cast<RankedTensorType>(); in GetSlicedBoundRanges() local2751 auto input_ty = nms_op.getType().template cast<ShapedType>(); in matchAndRewrite() local3500 const auto &input_ty = op.input().getType().cast<ShapedType>(); in verify() local3568 auto input_ty = op.operand().getType().cast<ShapedType>(); in verify() local3640 const auto &input_ty = op.input().getType(); in verify() local
1379 auto input_ty = op.x().getType().template dyn_cast<RankedTensorType>(); in Verify() local1733 const TensorType input_ty = input.getType().template cast<TensorType>(); in inferConvReturnTypeComponents() local1825 auto input_ty = input().getType().dyn_cast<TensorType>(); in GetOptimalLayout() local1908 auto input_ty = input().getType().dyn_cast<TensorType>(); in GetOptimalLayout() local2021 auto input_ty = op.x().getType().dyn_cast<RankedTensorType>(); in verify() local2499 auto input_ty = input.getType().dyn_cast<RankedTensorType>(); in InferExpandDimsOpType() local
64 Type input_ty = input.getType(); in InferReductionOpType() local
375 auto input_ty = input.getType().cast<ShapedType>(); in matchAndRewrite() local791 Type input_ty = input.getType(); in matchAndRewrite() local1062 auto input_ty = input.getType().cast<ShapedType>(); in matchAndRewrite() local1330 auto input_ty = input.getType().cast<ShapedType>(); in matchAndRewrite() local1547 auto input_ty = tf_roll_op.input().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local
1425 auto input_ty = op.input().getType().cast<ShapedType>(); in InferShapeForXlaReduceWindowOp() local2702 if (auto input_ty = func_type.getInput(i).dyn_cast<RankedTensorType>()) { in InferShapeForFunction() local
243 auto input_ty = input.getType().cast<ShapedType>(); in CanOptimizeIdentitySliceOp() local1364 auto input_ty = input.getType().dyn_cast<ShapedType>(); in matchAndRewrite() local1552 auto input_ty = value.getType().dyn_cast_or_null<ShapedType>(); in ComputeSliceK() local
78 PrimitiveType input_ty = instr->operand(0)->shape().element_type(); in HeuristicLayoutAssignment() local
63 PrimitiveType input_ty = instr->operand(0)->shape().element_type(); in GetRelevantConvs() local
1212 const RankedTensorType input_ty = in inferReturnTypes() local1758 auto input_ty = input.getType().cast<TensorType>(); in GetReshapeOutputType() local1860 auto input_ty = op.input().getType().cast<TensorType>(); in verify() local2144 auto input_ty = input().getType().dyn_cast<RankedTensorType>(); in fold() local
1981 auto input_ty = input.getType().cast<ShapedType>(); in matchAndRewrite() local2500 RankedTensorType input_ty = input.getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local2604 RankedTensorType input_ty = input.getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local