Searched refs:transform_shape (Results 1 – 1 of 1) sorted by relevance
179 SmallVector<int64_t, 4> transform_shape = { in HandlePad() local185 if (failed(HandleCast(cast_op, transform_shape))) return failure(); in HandlePad()188 RankedTensorType::get(transform_shape, getElementTypeOrSelf(input)); in HandlePad()210 SmallVector<int64_t, 4> transform_shape = { in HandleConv2DInput() local214 RankedTensorType::get(transform_shape, getElementTypeOrSelf(input)); in HandleConv2DInput()429 SmallVector<int64_t, 4> transform_shape = { in BuildSpaceToDepth() local433 RankedTensorType::get(transform_shape, getElementTypeOrSelf(input)); in BuildSpaceToDepth()