Home
last modified time | relevance | path

Searched refs:padded_shape_splits (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlower_tf.cc924 SmallVector<Value, 4> padded_shape_splits( in matchAndRewrite() local
944 loc, padded_shape_splits[1 + i], block_shape_splits[i])); in matchAndRewrite()
957 SmallVector<Value, 6> reshaped_shape_vals{padded_shape_splits[0]}; in matchAndRewrite()
967 reshaped_shape_vals.push_back(padded_shape_splits[i]); in matchAndRewrite()
993 auto output_batch = padded_shape_splits[0]; in matchAndRewrite()
1003 output_shape_vals.push_back(padded_shape_splits[i]); in matchAndRewrite()