Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlower_tf.cc880 int64_t paddings_sum = in matchAndRewrite() local
885 padded_shape[i + 1] = (paddings_sum + input_shape[i + 1]); in matchAndRewrite()
907 auto paddings_sum = rewriter.create<AddV2Op>( in matchAndRewrite() local
918 rewriter.create<AddV2Op>(loc, paddings_sum, input_shape_tensor); in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc1350 int64_t paddings_sum = in inferReturnTypes() local
1356 return_shape[1 + i] = (paddings_sum + input_shape[i + 1]) / block_shape_i; in inferReturnTypes()