Searched refs:dim_map_val (Results 1 – 1 of 1) sorted by relevance
1002 auto dim_map_val = instr_to_dim_map_[producer]; in PropagateOnBroadcast() local1004 const int64 old_batch_dim = dim_map_val.first; in PropagateOnBroadcast()1005 const int64 old_space_dim = dim_map_val.second; in PropagateOnBroadcast()1205 auto dim_map_val = instr_to_dim_map_[producer]; in Propagate() local1338 instr_to_dim_map_[consumer] = dim_map_val; in Propagate()1365 auto dim_map_val = instr_to_dim_map_[consumer->mutable_operand(0)]; in Propagate() local1366 const int64 old_batch_dim = dim_map_val.first; in Propagate()1367 const int64 old_space_dim = dim_map_val.second; in Propagate()1389 instr_to_dim_map_[consumer] = dim_map_val; in Propagate()1404 auto dim_map_val = instr_to_dim_map_[consumer->mutable_operand(0)]; in Propagate() local[all …]