Searched refs:out_width_constant (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | lower_tf.cc | 1321 int out_width_constant = -1; in matchAndRewrite() local
1330 out_width_constant = cst_size[1]; in matchAndRewrite()
1332 if (out_height_constant < 0 || out_width_constant < 0) return failure(); in matchAndRewrite()
1335 int out_spatial_cst = out_height_constant < 0 || out_width_constant < 0 in matchAndRewrite()
1337 : out_height_constant * out_width_constant; in matchAndRewrite()
1451 loc, RankedTensorType::get({out_width_constant}, rewriter.getF32Type()), in matchAndRewrite()
1455 loc, RankedTensorType::get({out_width_constant}, rewriter.getF32Type()), in matchAndRewrite()
1459 loc, RankedTensorType::get({out_width_constant}, out_size_element_ty), in matchAndRewrite()
1464 RankedTensorType::get({1, out_width_constant}, out_size_element_ty), in matchAndRewrite()
1473 RankedTensorType::get({out_height_constant, out_width_constant}, in matchAndRewrite()
|