Searched refs:out_size_element_ty (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | lower_tf.cc | 1306 auto out_size_element_ty = out_size_ty.getElementType(); in matchAndRewrite() local
1359 rewriter.create<ConstOp>(loc, GetScalarOfType(out_size_element_ty, 1)); in matchAndRewrite()
1365 loc, RankedTensorType::get({4}, out_size_element_ty), input_shape); in matchAndRewrite()
1367 auto scalar_dim_ty = RankedTensorType::get({}, out_size_element_ty); in matchAndRewrite()
1380 loc, RankedTensorType::get({}, out_size_element_ty), in_y, in_x); in matchAndRewrite()
1390 loc, RankedTensorType::get({}, out_size_element_ty), out_y, out_x); in matchAndRewrite()
1394 loc, RankedTensorType::get({4}, out_size_element_ty), in matchAndRewrite()
1423 loc, RankedTensorType::get({out_height_constant}, out_size_element_ty), in matchAndRewrite()
1428 RankedTensorType::get({out_height_constant, 1}, out_size_element_ty), in matchAndRewrite()
1431 RankedTensorType::get({2}, out_size_element_ty), in matchAndRewrite()
[all …]
|