Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_space_to_depth_pass.cc717 for (auto conv2d_and_block_size : conv2d_and_block_sizes) { in runOnOperation() local
718 if (conv2d_and_block_size.second != block_size) { in runOnOperation()
738 for (auto conv2d_and_block_size : conv2d_and_block_sizes) { in runOnOperation() local
739 HandleFirstConvolution(conv2d_and_block_size.first, in runOnOperation()
740 conv2d_and_block_size.second); in runOnOperation()