Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dspace_to_batch_converter.cc216 HloInstructionSet non_propagatable_instrs_; member in xla::__anon6418faad0111::ConvolutionVisitor
631 for (auto instr : non_propagatable_instrs_) { in Run()
667 non_propagatable_instrs_.clear(); in Run()
1833 non_propagatable_instrs_.erase(user); in PropagateOnUsers()
1837 non_propagatable_instrs_.insert(user); in PropagateOnUsers()
2817 if (non_propagatable_instrs_.count(convolution) > 0) { in PerformSpaceToBatchOnConvolution()
2818 non_propagatable_instrs_.erase(convolution); in PerformSpaceToBatchOnConvolution()