Searched refs:out_height_constant (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | lower_tf.cc | 1320 int out_height_constant = -1; in matchAndRewrite() local
1329 out_height_constant = cst_size[0]; 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()
1414 RankedTensorType::get({out_height_constant}, rewriter.getF32Type()), in matchAndRewrite()
1419 RankedTensorType::get({out_height_constant}, rewriter.getF32Type()), 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()
1436 RankedTensorType::get({out_height_constant, 1}, out_size_element_ty), in matchAndRewrite()
[all …]
|