Searched refs:dim_map_val (Results 1 – 1 of 1) sorted by relevance
358 auto dim_map_val = instr_to_dim_map_[old_operand]; in GetSpatialDimsToSplit() local363 old_dims[0] = dim_map_val[DimMapper(SpaceToBatchDimMap::kSpace0)]; in GetSpatialDimsToSplit()1410 auto dim_map_val = instr_to_dim_map_[producer]; in PropagateOnBroadcast() local1413 dim_map_val[DimMapper(SpaceToBatchDimMap::kBatch)]; in PropagateOnBroadcast()1415 dim_map_val[DimMapper(SpaceToBatchDimMap::kSpace0)]; in PropagateOnBroadcast()1740 auto dim_map_val = instr_to_dim_map_[producer]; in Propagate() local1875 instr_to_dim_map_[consumer] = std::vector<int64_t>(dim_map_val); in Propagate()1919 auto dim_map_val = instr_to_dim_map_[consumer->mutable_operand(0)]; in Propagate() local1921 dim_map_val[DimMapper(SpaceToBatchDimMap::kBatch)]; in Propagate()1947 instr_to_dim_map_[consumer] = std::vector<int64_t>(dim_map_val); in Propagate()[all …]