Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_space_to_depth_pass.cc701 auto conv2d_and_block_sizes = argnum_and_convolution.getSecond(); in runOnOperation() local
705 if (argnum_num_users[arg_num] != conv2d_and_block_sizes.size() || in runOnOperation()
706 conv2d_and_block_sizes.empty()) { in runOnOperation()
710 int64_t block_size = conv2d_and_block_sizes[0].second; in runOnOperation()
717 for (auto conv2d_and_block_size : conv2d_and_block_sizes) { in runOnOperation()
732 auto conv2d_and_block_sizes = argnum_and_convolution.getSecond(); in runOnOperation() local
733 int64_t block_size = conv2d_and_block_sizes[0].second; in runOnOperation()
738 for (auto conv2d_and_block_size : conv2d_and_block_sizes) { in runOnOperation()