Searched refs:instr_to_dim_map_ (Results 1 – 1 of 1) sorted by relevance
302 absl::flat_hash_map<HloInstruction*, std::vector<int64_t>> instr_to_dim_map_; member in xla::__anon5e1d62920111::ConvolutionVisitor358 auto dim_map_val = instr_to_dim_map_[old_operand]; in GetSpatialDimsToSplit()488 auto dim_map_val_op_0 = instr_to_dim_map_[kernel]; in IsThisBackPropFilterConv()498 auto dim_map_val_op_0 = instr_to_dim_map_[activations]; in IsThisBackPropFilterConv()983 CHECK(instr_to_dim_map_.contains(old_producer)); in CanPropagate()988 if (instr_to_dim_map_[pivot_operand] in CanPropagate()990 instr_to_dim_map_[old_producer] in CanPropagate()992 instr_to_dim_map_[pivot_operand] in CanPropagate()994 instr_to_dim_map_[old_producer] in CanPropagate()1019 j == instr_to_dim_map_[pivot_operand][DimMapper( in CanPropagate()[all …]