Searched refs:batch_to_space_map_ (Results 1 – 1 of 1) sorted by relevance
219 absl::flat_hash_map<HloInstruction*, HloInstruction*> batch_to_space_map_; member in xla::__anon6418faad0111::ConvolutionVisitor1708 if (batch_to_space_map_.count(old_instr)) { in BatchToSpace()1709 CHECK_NE(batch_to_space_map_[old_instr], nullptr); in BatchToSpace()1710 return batch_to_space_map_[old_instr]; in BatchToSpace()1756 batch_to_space_map_[old_instr] = output_transpose; in BatchToSpace()2788 batch_to_space_map_[convolution->mutable_operand(0)] = in PerformSpaceToBatchOnConvolution()