Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dtpu_space_to_depth_pass.cc654 for (auto argnum_and_convolution : argnum_and_convolutions) { in runOnOperation() local
655 auto arg_num = argnum_and_convolution.getFirst(); in runOnOperation()
656 auto conv2d_and_block_sizes = argnum_and_convolution.getSecond(); in runOnOperation()
686 for (auto argnum_and_convolution : argnum_and_convolutions) { in runOnOperation() local
687 auto conv2d_and_block_sizes = argnum_and_convolution.getSecond(); in runOnOperation()
691 argnum_and_convolution.getFirst()); in runOnOperation()