Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_space_to_depth_pass.cc669 argnum_and_convolutions; in runOnOperation() local
682 argnum_and_convolutions[arg_num].emplace_back(conv2d, block_size); in runOnOperation()
699 for (auto argnum_and_convolution : argnum_and_convolutions) { in runOnOperation()
707 argnum_and_convolutions.erase(arg_num); in runOnOperation()
712 argnum_and_convolutions.erase(arg_num); in runOnOperation()
719 argnum_and_convolutions.erase(arg_num); in runOnOperation()
726 if (argnum_and_convolutions.empty()) { in runOnOperation()
731 for (auto argnum_and_convolution : argnum_and_convolutions) { in runOnOperation()